gpt4 book ai didi

python - kedro 中是否有 IO 功能来存储训练好的模型?

转载 作者:行者123 更新时间:2023-12-03 08:45:15 25 4
gpt4 key购买 nike

IO section of the kedro API docs我找不到相关功能。存储经过训练的模型(例如 .pkl.joblibONNXPMML)?我错过了什么吗?

最佳答案

pickle kedro.io 中的数据集,您可以使用它来保存经过训练的模型和/或您想要腌制的任何内容,并且是可序列化的(模型是通用对象)。它接受默认为 picklebackend,但如果您想使用 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/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com