如何使用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 文件的链接以及配置字典。