python - 如何使用 TRAINS python 自动魔法实验管理器手动注册 sci-kit 模型？

Question

我主要使用 scikit-learn，据我了解，TRAINS 自动魔术不会自动捕获 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 文件的链接以及配置字典。