Ver código fonte

modify: jupyter输出优化

yq 3 meses atrás
pai
commit
cebaf63d4e
1 arquivos alterados com 3 adições e 1 exclusões
  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__":