python - 是否有 IO 功能可以在 kedro 中存储经过训练的模型？

Question

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

标签： pythonkedro

Answer 1

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

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

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