Browse Source

modify: jupyter输出优化

yq 3 tháng trước cách đây
mục cha
commit
cebaf63d4e
1 tập tin đã thay đổi với 3 bổ sung1 xóa
  1. 3 1
      trainer/train.py

+ 3 - 1
trainer/train.py

@@ -35,8 +35,10 @@ class TrainPipeline():
         return self.metric_value_dict
 
     def generate_report(self, ):
+        save_path = self._filter_strategy.data_process_config.f_get_save_path("模型报告.docx")
         Report.generate_report(self.metric_value_dict, self._model.get_template_path(),
-                               save_path=self._filter_strategy.data_process_config.f_get_save_path("模型报告.docx"))
+                               save_path=save_path)
+        print(f"模型报告文件储存路径:{save_path}")
 
 
 if __name__ == "__main__":