Эх сурвалжийг харах

bugfix: metric_value_dict

yq 4 сар өмнө
parent
commit
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(),