Browse Source

bugfix: metric_value_dict

yq 4 months ago
parent
commit
effbc96070
1 changed files with 1 additions and 1 deletions
  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(),