我尝试同时使用mlflow.keras.autolog
和mlflow.log_param
(mlflow v 1.22.0
)记录我的ML试验。但是,唯一记录的东西是autolog
的产品,而不是log_param
的产品。
experiment = mlf_client.get_experiment_by_name(experiment_name)
with mlflow.start_run(experiment_id=experiment.experiment_id):
mlflow.keras.autolog(log_input_examples=True)
mlflow.log_param('batch_size', self.batch_size)
mlflow.log_param('training_set_size', len(kwargs['training_ID_list']))
mlflow.log_param('testing_set_size', len(kwargs['testing_ID_list']))
history = self.train_NN_model(**kwargs)
我知道我可以使用log_param
和log_model
来保存模型本身,但是这样我就丢失了一些autolog
可以自动为我记录的有用内容(例如,模型摘要)。
是否可以使用**autolog
和自定义**参数进行日志记录?
1条答案
按热度按时间qvtsj1bj1#
是的,你可以。让我通过改编mlflow文档中的示例来演示这一点
请参阅我作为要点分享的this fully working example。