如何使用TRAINS python auto-magical實驗管理器手動注冊sci-kit模型？

Question

我主要使用 scikit-learn，據我所知，TRAINS auto-magic 不會自動捕獲 scikit-learn 模型存儲/加載。

我如何在“腌制”模型后手動注冊模型。

例如：

import pickle
with open("model.pkl", "wb") as file:  
    pickle.dump(my_model, file)

Answer 1

假設您指的是 TRAINS 實驗經理： https : //github.com/allegroai/trains （我是維護者之一）

from trains import Task, OutputModel
OutputModel(Task.current_task()).update_weights(weights_filename="model.pkl")

或者，如果您有想要與腌制模型文件一起存儲的信息，您可以執行以下操作：

from trains import Task, OutputModel
model_parameters = {'threshold': 0.123}
OutputModel(Task.current_task(), config_dict=model_parameters).update_weights(weights_filename="model.pkl")

現在，您應該在 UI 中看到一個在實驗中注冊的輸出模型。 該模型包含指向pickel 文件的鏈接以及配置字典。