StratifiedKFold：隨機播放和random_state

Question

為了進行交叉驗證，我已經完成了以下操作：

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

我只是在考慮一個細節：如果有人（例如我的老師！）再次運行該代碼，我希望得到相同的最終結果。 但是，我忘了指定random_state參數！ 不幸的是，我無法從頭開始，因為我的模型需要很長時間才能安裝，並且已經完成。

我的問題如下：是否可以找到導致我的SKFolds的random_state是什么？ （我的筆記本仍在打開，所以也許可以在某處找到該信息？）。 還是我可以做一些類似的事情，例如，將我的SKFolds保存到一個csv文件中，然后在重新啟動筆記本電腦時加載它，以確保折疊時得到相同的拆分？

謝謝你的幫助！

Answer 1

您可以將SKFolds對象與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" ) )