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

原油测试集预测停止，不生成报告，不评估模型

Browse Source
This commit is contained in:
workpc 2025-07-07 17:16:44 +08:00
parent 4c5a1d1c6e
commit 91387a7241
2 changed files with 21 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
main_yuanyou_yuedu.py
View File

@ -543,25 +543,25 @@ def predict_main():
logger.info('模型训练完成')
logger.info('训练数据绘图ing')
model_results3 = model_losss(sqlitedb, end_time=end_time)
logger.info('训练数据绘图end')
# logger.info('训练数据绘图ing')
# model_results3 = model_losss(sqlitedb, end_time=end_time)
# logger.info('训练数据绘图end')
push_market_value()
# push_market_value()
# 模型报告
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('模型训练完成')
sql_inset_predict(global_config)

10
models/nerulforcastmodels.py
View File

@ -212,11 +212,11 @@ def ex_Model(df, horizon, input_size, train_steps, val_check_steps, early_stop_p
config.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(
config.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(
# config.dataset, "cross_validation.csv"), index=False)
df_test['ds'] = pd.to_datetime(df_test['ds'], errors='coerce')