sk-learn保存模型到磁盘,但只获取数组

时间:2017-07-28 05:10:40

标签: python numpy scikit-learn pickle joblib

存储fitted_clf sk-learn分类器时:

joblib.dump(fitted_clf, some_path)

大部分时间将其重新加载到内存中,如:

joblib.load(some_path)

只返回一个array(['col1', 'col2], dtype=object)数组,而不是加载已安装的管道。

然而,有时我会得到真正的管道,但不明白为什么这不是一致的行为。

修改

我认为这与不同的joblib判断有关,即 从sklearn.externals import joblib起作用,但在使用常规joblib时我只得到一个数组

1 个答案:

答案 0 :(得分:2)

确认。使用sklearn.externals import joblib正在修复此问题以保持一致的行为。