分层折叠:随机播放和随机播放

2024-03-29 00:28:08 发布

您现在位置:Python中文网/ 问答频道 /正文

为了交叉验证,我做了以下工作:

from sklearn.cross_validation import StratifiedKFold
n_folds = 5
SKFolds = list(StratifiedKFold(ytrain, n_folds, shuffle=True))

我只想了一个细节:我希望有人(比如我的老师)有同样的最终结果再次运行代码。但是,我忘了指定random_state参数!不幸的是,我不能从头开始,因为我的模型需要很长的时间来安装,而且已经完成了。在

我的问题是:有没有可能找到导致我的SKFolds的随机_状态?(我的笔记本还在打开,所以也许可以在某处找到信息?)。或者我可以做一些事情,比如把SKFolds保存在csv文件中,然后在我重新启动笔记本时加载它,以确保我的折叠上也有相同的拆分?在

谢谢你的帮助!在


Tags: fromimporttrue笔记本sklearn交叉细节list
1条回答
网友
1楼 · 发布于 2024-03-29 00:28:08

您可以用pickle保存SKFolds对象,然后只需加载它并按原样使用它。在

import cPickle as pickle

# To save the object
pickle.dump( SKFolds , open( "skfolds.p", "wb" ) )

# To load the object
SKFolds = pickle.load( open( "skfolds.p", "rb" ) )

相关问题 更多 >