浏览代码

bugfix: metric_value_dict

yq 4 月之前
父节点
当前提交
effbc96070
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      trainer/train.py

+ 1 - 1
trainer/train.py

@@ -27,7 +27,7 @@ class TrainPipeline():
         metric_value_dict_feature = self._filter_strategy.feature_report(self._data, candidate_feature)
 
         metric_value_dict_train = self._model.train(data_prepared, *data_prepared.args, **data_prepared.kwargs)
-        self.metric_value_dict = metric_value_dict_feature.update(metric_value_dict_train)
+        self.metric_value_dict = {**metric_value_dict_feature, **metric_value_dict_train}
 
     def generate_report(self, ):
         Report.generate_report(self.metric_value_dict, self._model.get_template_path(),