如何在数据块中保存和重用我的模型
How to save and reuse my model in databricks
我已经准备好从数据输入到最终预测的完整管道。我有 10 个模型(编写了使整个过程自动化的函数)然后,根据准确性选择 1 个模型。现在,我需要存储所有模型(因为随着数据的变化,任何模型都可以具有更高的准确性)并使用来自我的训练特征的新输入值重用它。
由于所有过程都非常动态,因此每个步骤作业都会触发并完成作业。在如何保存和重用与此流程一致的模型方面需要帮助。
提前谢谢你。举个例子会有很大帮助
你尝试了吗?
mlflow.pyfunc.log_model for saving the model
mlflow.pyfunc.load_model(f"models:/{model_name}/production") for loading the model
我已经准备好从数据输入到最终预测的完整管道。我有 10 个模型(编写了使整个过程自动化的函数)然后,根据准确性选择 1 个模型。现在,我需要存储所有模型(因为随着数据的变化,任何模型都可以具有更高的准确性)并使用来自我的训练特征的新输入值重用它。
由于所有过程都非常动态,因此每个步骤作业都会触发并完成作业。在如何保存和重用与此流程一致的模型方面需要帮助。
提前谢谢你。举个例子会有很大帮助
你尝试了吗?
mlflow.pyfunc.log_model for saving the model
mlflow.pyfunc.load_model(f"models:/{model_name}/production") for loading the model