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-02-12 08:49:44 +08:00
parent c29bf2891b
commit 7b39b67842
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
config_jingbo_zhoudu.py
View File

@ -159,9 +159,9 @@ table_name = 'v_tbl_crude_oil_warning'
### 开关
is_train = True # 是否训练
is_train = False # 是否训练
is_debug = False # 是否调试
is_eta = True # 是否使用eta接口
is_eta = False # 是否使用eta接口
is_market = True # 是否通过市场信息平台获取特征 ,在is_eta 为true 的情况下生效
is_timefurture = True # 是否使用时间特征
is_fivemodels = False # 是否使用之前保存的最佳的5个模型

6
models/nerulforcastmodels.py
View File

@ -188,9 +188,9 @@ def ex_Model(df,horizon,input_size,train_steps,val_check_steps,early_stop_patien
logger.info('读取模型：'+ filename)
nf = load(filename)
# # 测试集预测
# nf_test_preds = nf.cross_validation(df=df_test, val_size=val_size, test_size=test_size, n_windows=None)
# # 测试集预测结果保存
# nf_test_preds.to_csv(os.path.join(dataset,"cross_validation.csv"),index=False)
nf_test_preds = nf.cross_validation(df=df_test, val_size=val_size, test_size=test_size, n_windows=None)
# 测试集预测结果保存
nf_test_preds.to_csv(os.path.join(dataset,"cross_validation.csv"),index=False)
df_test['ds'] = pd.to_datetime(df_test['ds'], errors='coerce')