2

在kedro API 文档的 IO 部分中，我找不到存储训练模型的功能（例如.pkl，、、、.joblib）？我错过了什么吗？ONNXPMML

4

2 回答 2

3

中有pickle数据集kedro.io，您可以使用它来保存经过训练的模型和/或您想要腌制的任何东西，并且是可序列化的（模型是一个常见的对象）。它接受backend默认为的 a，但如果您想改用，pickle可以设置为。joblibjoblib

我只是要快速指出，Kedro 正在将kedro.extras.datasets其数据集转移到kedro.io. 您可能希望查看kedro.extras.datasetsKedro 0.16 及更高版本pickle.PickleDataSet的joblib支持。

如果您想查看示例，文档中的 Kedrospaceflights教程实际上使用数据集保存了经过训练的线性回归模型。pickle相关部分在这里。

于 2020-05-11T19:38:39.163 回答
1

在https://kedro.readthedocs.io/en/latest/kedro.extras.datasets.pickle.PickleDataSet.html中有 PickleDataSet ，PickleDataSet 中的 joblib 支持在下一版本中（参见https://github.com/quantumblacklabs /kedro/blob/develop/RELEASE.md )

于 2020-05-11T19:37:33.130 回答