gpt4 book ai didi

python - 如何在 Databricks 中检索实验中的 model.pkl

转载 作者:行者123 更新时间:2023-12-02 18:36:16 29 4
gpt4 key购买 nike

我想从我的训练模型中检索 pickle,我知道它位于 Databricks 实验中的运行文件中。

看来mlflow.pyfunc.load_model只能执行predict方法。

有一个选项可以直接访问pickle吗?

我还尝试使用 pickle.load(path) 在运行中使用该路径(路径示例:dbfs:/databricks/mlflow-tracking/20526156406/92f3ec23bf614c9d934dd0195/artifacts/model/模型.pkl)。

最佳答案

使用frmwk的原生load_model()方法(例如sklearn.load_model())或download_artifacts()

关于python - 如何在 Databricks 中检索实验中的 model.pkl,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/68718719/

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