如何在 databricks 中保存和重復使用我的 model

Question

我擁有從數據輸入到最終預測的完整管道。 我有10個模型（寫了function來自動化整個過程）然后，根據准確性選擇1 model。 現在，我需要存儲所有模型（因為任何 model 都可以隨着數據的變化而具有更好的准確性）並使用來自我的火車特征的新輸入值重用它。

由於所有過程都是非常動態的，因此每一步作業都會觸發並完成作業。 在如何保存和重用與此流程一致的 model 方面需要幫助。

先感謝您。 一個例子會很有幫助

Answer 1

你試過了嗎？

mlflow.pyfunc.log_model  for saving the model
mlflow.pyfunc.load_model(f"models:/{model_name}/production") for loading the model

mlflow.pyfunc