From b2c958a4fd7b432756d6b372c2e4fb9758c59abd Mon Sep 17 00:00:00 2001
From: workpc <liurui_test@163.com>
Date: Tue, 24 Jun 2025 16:12:25 +0800
Subject: [PATCH] =?UTF-8?q?=E6=89=A7=E8=A1=8C=E9=80=BB=E8=BE=91=E8=B0=83?=
 =?UTF-8?q?=E6=95=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main_yuanyou.py        | 33 ++++++++++++++++-----------------
 main_yuanyou_yuedu.py  |  5 +++--
 main_yuanyou_zhoudu.py |  4 ++--
 3 files changed, 21 insertions(+), 21 deletions(-)

diff --git a/main_yuanyou.py b/main_yuanyou.py
index ab150d5..85ab106 100644
--- a/main_yuanyou.py
+++ b/main_yuanyou.py
@@ -516,23 +516,23 @@ def predict_main():
     model_results3 = model_losss(sqlitedb, end_time=end_time)
     logger.info('训练数据绘图end')
 
-    # # 模型报告
-    # logger.info('制作报告ing')
-    # title = f'{settings}--{end_time}-预测报告'  # 报告标题
-    # reportname = f'Brent原油大模型日度预测--{end_time}.pdf'  # 报告文件名
-    # reportname = reportname.replace(':', '-')  # 替换冒号
-    # brent_export_pdf(dataset=dataset,
-    #                  num_models=5 if is_fivemodels else 22, time=end_time,
-    #                  reportname=reportname,
-    #                  inputsize=global_config['horizon'],
-    #                  sqlitedb=sqlitedb
-    #                  ),
+    # 模型报告
+    logger.info('制作报告ing')
+    title = f'{settings}--{end_time}-预测报告'  # 报告标题
+    reportname = f'Brent原油大模型日度预测--{end_time}.pdf'  # 报告文件名
+    reportname = reportname.replace(':', '-')  # 替换冒号
+    brent_export_pdf(dataset=dataset,
+                     num_models=5 if is_fivemodels else 22, time=end_time,
+                     reportname=reportname,
+                     inputsize=global_config['horizon'],
+                     sqlitedb=sqlitedb
+                     ),
 
-    # logger.info('制作报告end')
-    # logger.info('模型训练完成')
+    logger.info('制作报告end')
+    logger.info('模型训练完成')
 
-    # push_market_value()
-    # sql_inset_predict(global_config)
+    push_market_value()
+    sql_inset_predict(global_config)
 
     # # LSTM 单变量模型
     # ex_Lstm(df,input_seq_len=input_size,output_seq_len=horizon,is_debug=is_debug,dataset=dataset)
@@ -563,6 +563,5 @@ if __name__ == '__main__':
     #     end_time = i_time.strftime('%Y-%m-%d')
     #     predict_main()
 
-    # predict_main()
+    predict_main()
     # push_market_value()
-    sql_inset_predict(global_config)
diff --git a/main_yuanyou_yuedu.py b/main_yuanyou_yuedu.py
index 1702fa1..a25f0c8 100644
--- a/main_yuanyou_yuedu.py
+++ b/main_yuanyou_yuedu.py
@@ -473,6 +473,8 @@ def predict_main():
     logger.info('制作报告end')
     logger.info('模型训练完成')
 
+    sql_inset_predict(global_config)
+
     # # LSTM 单变量模型
     # ex_Lstm(df,input_seq_len=input_size,output_seq_len=horizon,is_debug=is_debug,dataset=dataset)
 
@@ -506,5 +508,4 @@ if __name__ == '__main__':
     #         logger.info(f'预测失败：{e}')
     #         continue
 
-    # predict_main()
-    sql_inset_predict(global_config)
+    predict_main()
diff --git a/main_yuanyou_zhoudu.py b/main_yuanyou_zhoudu.py
index 2a56853..449d187 100644
--- a/main_yuanyou_zhoudu.py
+++ b/main_yuanyou_zhoudu.py
@@ -452,6 +452,7 @@ def predict_main():
     logger.info('模型训练完成')
 
     push_market_value()
+    sql_inset_predict(global_config)
 
     # 发送邮件
     # m = SendMail(
@@ -477,5 +478,4 @@ if __name__ == '__main__':
     #         logger.info(f'预测失败：{e}')
     #         continue
 
-    # predict_main()
-    sql_inset_predict(global_config)
+    predict_main()