liurui/PriceForecast
Template
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

聚烯烃参数名称

Browse Source
This commit is contained in:
jingboyitiji 2025-03-28 11:27:39 +08:00
parent 5a51fecbde
commit b9b23dbf77
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main_juxiting.py
View File

@ -3,7 +3,7 @@
from lib.dataread import * from lib.dataread import *
from config_juxiting import * from config_juxiting import *
from lib.tools import SendMail, exception_logger 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 datetime
import torch import torch
torch.set_float32_matmul_precision("high") torch.set_float32_matmul_precision("high")
@ -317,7 +317,7 @@ def predict_main():
row, col = df.shape row, col = df.shape
now = datetime.datetime.now().strftime('%Y%m%d%H%M%S') now = datetime.datetime.now().strftime('%Y%m%d%H%M%S')
ex_Model(df, ex_Model_Juxiting(df,
horizon=global_config['horizon'], horizon=global_config['horizon'],
input_size=global_config['input_size'], input_size=global_config['input_size'],
train_steps=global_config['train_steps'], train_steps=global_config['train_steps'],

2
models/nerulforcastmodels.py
View File

@ -404,7 +404,7 @@ def ex_Model_Juxiting(df, horizon, input_size, train_steps, val_check_steps, ear
models.append(model) models.append(model)
# 创建NeuralForecast实例并训练模型 # 创建NeuralForecast实例并训练模型
nf = NeuralForecast(models=models, freq=freq) nf = NeuralForecast(models=models, freq=config.freq)
from joblib import dump, load from joblib import dump, load
if is_train: if is_train: