【发布时间】:2018-06-06 02:56:03
【问题描述】:
有没有在 python 中保存 OneHotencoder 对象? .原因是我在训练数据和测试数据的预处理中使用了该对象,我们正在构建一个包含相同训练模型的 API,并且在用户创建时将由来自网站的真实数据注入。所以首先需要对数据进行预处理,然后模型可以预测相同的 o/p。
谢谢
【问题讨论】:
-
你用过
sklearn的joblib.dump和load函数吗? -
由于我使用的是 keras 模型,我可以保存模型并减轻重量。问题在于数据的预处理。我想保留我的 OHE 对象,以便我可以转换看不见的数据。
-
预处理器是否是 keras 的一部分?如果是 scikit 模型,请阅读:scikit-learn.org/stable/modules/model_persistence.html
-
是的,它有帮助。在 OHE 中简单地使用 pickel 就可以了。
-
我推荐了
joblib,第二个例子,因为它有一些优点(例如并行化)。
标签: python scikit-learn one-hot-encoding