在 IO section of the kedro API docs我找不到相关功能。存储经过训练的模型(例如 .pkl、.joblib、ONNX、PMML)？我错过了什么吗？
最佳答案
有pickle kedro.io 中的数据集，您可以使用它来保存经过训练的模型和/或您想要腌制的任何内容，并且是可序列化的(模型是通用对象)。它接受默认为 pickle 的 backend，但如果您想使用 joblib，可以将其设置为 joblib。
我只是想快速指出 Kedro 正在迁移到 kedro.extras.datasets其数据集，并摆脱 kedro.io 中的非核心数据集。您可能需要查看 kedro.extras.datasets 以及 Kedro 0.16 及以上 pickle.PickleDataSet具有 joblib 支持。
凯德罗 spaceflights如果您想查看示例，文档中的教程实际上使用 pickle 数据集保存经过训练的线性回归模型。相关部分是here .
关于python - kedro 中是否有 IO 功能来存储训练好的模型？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/61737613/