Как вручную зарегистрировать модель sci-kit с помощью менеджера авто-магических экспериментов TRAINS python?

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")

Теперь вы должны увидеть в интерфейсе выходную модель, зарегистрированную в эксперименте.Модель содержит ссылку на файл pickel вместе со словарем конфигурации.