2017-07-28 53 views
2

存儲當fitted_clf SK-學習分類,如:SK-學會保存模型到磁盤上,但只得到陣列

joblib.dump(fitted_clf, some_path) 

加載它當回內存最喜歡的時間:

joblib.load(some_path) 

只返回array(['col1', 'col2], dtype=object)的一個數組而不是加載擬合管道。

但是,有時我會得到真正的流水線,但不明白爲什麼這不是一個一致的行爲。

編輯

我認爲這與不同JOBLIB verisions即 從sklearn.externals import joblib工作要做,但使用常規joblib時,我只得到一個數組

回答

2

證實。使用sklearn.externals import joblib正在修復此問題以具有一致的行爲。