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

Merge branch 'main' of https://gitea.jayhgq.cn/liurui/PriceForecast

Browse Source
This commit is contained in:
workpc 2025-08-05 17:39:01 +08:00
commit 3b51565262
3 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lib/tools.py
View File

@ -780,6 +780,7 @@ def get_model_id_name_dict(global_config):
'''
tb = 'v_tbl_predict_models'
sql = f'select model_name,id from {tb} '
global_config['db_mysql'].connect()
modelsname = global_config['db_mysql'].execute_query(sql)
model_id_name_dict = {row['id']: row['model_name'] for row in modelsname}
global_config['logger'].info(f'模型id-name: {model_id_name_dict}')
@ -1123,7 +1124,7 @@ def plot_pp_predict_result(y_hat, global_config,wd='yuedu'):
sns.lineplot(x=y['ds'], y=y['y'], color='blue', label='真实值', ax=ax)
# date_str = pd.Timestamp(y_hat["ds"].iloc[0]).strftime('%Y-%m-%d')
ax.set_title(f'{global_config["end_time"]} PP期货八大维度 预测价格走势')
ax.set_title(f'{global_config["end_time"]} PP期货八大维度 预测价格走势', fontsize=24)
ax.set_xlabel('日期')
ax.set_ylabel('预测结果')
ax.tick_params(axis='x', rotation=45)
@ -1147,6 +1148,10 @@ def plot_pp_predict_result(y_hat, global_config,wd='yuedu'):
# 在图表下方添加表格
table = ax.table(cellText=data, colLabels=columns,
loc='bottom', bbox=[0, -0.4, 1, 0.2],cellLoc='center')
# 设置表头为浅蓝色
for (i, j), cell in table.get_celld().items():
if i == 0: # 表头行
cell.set_facecolor('lightblue')
table.auto_set_font_size(False)
table.set_fontsize(12)
@ -1161,6 +1166,10 @@ def plot_pp_predict_result(y_hat, global_config,wd='yuedu'):
bbox=[0, -0.9, 1, 0.4], # [左, 下, 宽, 高]，调整下边界(-0.7)和高度(0.5)控制表格位置
cellLoc='left'
)
# 设置表头为浅蓝色
for (i, j), cell in table.get_celld().items():
if i == 0: # 表头行
cell.set_facecolor('lightblue')
table.auto_set_font_size(False) # 关闭自动字体大小
table.set_fontsize(10) # 设置表格字体大小
table.scale(1.1, 1.5) # 调整表格缩放比例（宽度, 高度）

3
main_juxiting_yuedu.py
View File

@ -477,7 +477,7 @@ def predict_main():
# sqlitedb, end_time=global_config['end_time'], is_fivemodels=global_config['is_fivemodels'])
# logger.info('训练数据绘图end')
push_market_value()
# push_market_value()
# sql_inset_predict(global_config)
@ -530,6 +530,7 @@ if __name__ == '__main__':
# logger.info(f'预测失败：{e}')
# continue
global_config['end_time'] = '2025-08-01'
predict_main()
# push_market_value()

4
main_juxiting_zhoudu.py
View File

@ -512,7 +512,11 @@ if __name__ == '__main__':
# logger.info(f'预测失败：{e}')
# continue
global_config['end_time'] = '2025-08-01'
global_config['db_mysql'].connect()
predict_main()
# push_market_value()
# sql_inset_predict(global_config)