浏览代码

modify: jupyter输出优化

yq 3 月之前
父节点
当前提交
cebaf63d4e
共有 1 个文件被更改,包括 3 次插入1 次删除
  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__":