为了交叉验证,我做了以下工作:
from sklearn.cross_validation import StratifiedKFold
n_folds = 5
SKFolds = list(StratifiedKFold(ytrain, n_folds, shuffle=True))
我只想了一个细节:我希望有人(比如我的老师)有同样的最终结果再次运行代码。但是,我忘了指定random_state参数!不幸的是,我不能从头开始,因为我的模型需要很长的时间来安装,而且已经完成了。在
我的问题是:有没有可能找到导致我的SKFolds的随机_状态?(我的笔记本还在打开,所以也许可以在某处找到信息?)。或者我可以做一些事情,比如把SKFolds保存在csv文件中,然后在我重新启动笔记本时加载它,以确保我的折叠上也有相同的拆分?在
谢谢你的帮助!在
您可以用
pickle
保存SKFolds
对象,然后只需加载它并按原样使用它。在相关问题 更多 >
编程相关推荐