From b9b23dbf77e42b1162b51c8f78a5b6a01c0aab0f Mon Sep 17 00:00:00 2001
From: jingboyitiji <jingboyitiji@admin.com>
Date: Fri, 28 Mar 2025 11:27:39 +0800
Subject: [PATCH] =?UTF-8?q?=E8=81=9A=E7=83=AF=E7=83=83=E5=8F=82=E6=95=B0?=
 =?UTF-8?q?=E5=90=8D=E7=A7=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main_juxiting.py             | 4 ++--
 models/nerulforcastmodels.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/main_juxiting.py b/main_juxiting.py
index 2e50c57..b63d2cd 100644
--- a/main_juxiting.py
+++ b/main_juxiting.py
@@ -3,7 +3,7 @@
 from lib.dataread import *
 from config_juxiting import *
 from lib.tools import SendMail, exception_logger
-from models.nerulforcastmodels import ex_Model, model_losss_juxiting, tansuanli_export_pdf, pp_export_pdf
+from models.nerulforcastmodels import ex_Model_Juxiting, model_losss_juxiting, pp_export_pdf
 import datetime
 import torch
 torch.set_float32_matmul_precision("high")
@@ -317,7 +317,7 @@ def predict_main():
     row, col = df.shape
 
     now = datetime.datetime.now().strftime('%Y%m%d%H%M%S')
-    ex_Model(df,
+    ex_Model_Juxiting(df,
              horizon=global_config['horizon'],
              input_size=global_config['input_size'],
              train_steps=global_config['train_steps'],
diff --git a/models/nerulforcastmodels.py b/models/nerulforcastmodels.py
index 870b8b3..72c9e25 100644
--- a/models/nerulforcastmodels.py
+++ b/models/nerulforcastmodels.py
@@ -404,7 +404,7 @@ def ex_Model_Juxiting(df, horizon, input_size, train_steps, val_check_steps, ear
                 models.append(model)
 
     # 创建NeuralForecast实例并训练模型
-    nf = NeuralForecast(models=models, freq=freq)
+    nf = NeuralForecast(models=models, freq=config.freq)
 
     from joblib import dump, load
     if is_train: