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: