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

报告上传接口添加 groupNo 字段

Browse Source
This commit is contained in:
workpc 2025-02-18 09:55:38 +08:00
parent aded91c23b
commit 4f2a888dff
4 changed files with 34 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
config_jingbo.py
View File

@ -112,6 +112,7 @@ login_data = {
} }
upload_data = { upload_data = {
"groupNo":'', # 用户组id
"funcModule":'研究报告信息', "funcModule":'研究报告信息',
"funcOperation":'上传原油价格预测报告', "funcOperation":'上传原油价格预测报告',
"data":{ "data":{
@ -129,6 +130,7 @@ upload_data = {
warning_data = { warning_data = {
"groupNo":'', # 用户组id
"funcModule":'原油特征停更预警', "funcModule":'原油特征停更预警',
"funcOperation":'原油特征停更预警', "funcOperation":'原油特征停更预警',
"data":{ "data":{
@ -161,7 +163,7 @@ table_name = 'v_tbl_crude_oil_warning'
### 开关 ### 开关
is_train = False # 是否训练 is_train = False # 是否训练
is_debug = False # 是否调试 is_debug = False # 是否调试
is_eta = False # 是否使用eta接口 is_eta = True # 是否使用eta接口
is_market = True # 是否通过市场信息平台获取特征 ,在is_eta 为true 的情况下生效 is_market = True # 是否通过市场信息平台获取特征 ,在is_eta 为true 的情况下生效
is_timefurture = True # 是否使用时间特征 is_timefurture = True # 是否使用时间特征
is_fivemodels = False # 是否使用之前保存的最佳的5个模型 is_fivemodels = False # 是否使用之前保存的最佳的5个模型

2
config_jingbo_pro.py
View File

@ -114,6 +114,7 @@ upload_data = {
"funcModule":'研究报告信息', "funcModule":'研究报告信息',
"funcOperation":'上传原油价格预测报告', "funcOperation":'上传原油价格预测报告',
"data":{ "data":{
"groupNo":'', # 用户组id
"ownerAccount":'27663', #报告所属用户账号 27663 - 刘小朋 "ownerAccount":'27663', #报告所属用户账号 27663 - 刘小朋
"reportType":'OIL_PRICE_FORECAST', # 报告类型，固定为OIL_PRICE_FORECAST "reportType":'OIL_PRICE_FORECAST', # 报告类型，固定为OIL_PRICE_FORECAST
"fileName": '', #文件名称 "fileName": '', #文件名称
@ -127,6 +128,7 @@ upload_data = {
} }
warning_data = { warning_data = {
"groupNo":'', # 用户组id
"funcModule":'原油特征停更预警', "funcModule":'原油特征停更预警',
"funcOperation":'原油特征停更预警', "funcOperation":'原油特征停更预警',
"data":{ "data":{

6
main_yongan.py
View File

@ -242,9 +242,9 @@ def predict_main():
# logger.info('模型训练完成') # logger.info('模型训练完成')
# logger.info('训练数据绘图ing') logger.info('训练数据绘图ing')
# model_results3 = model_losss_yongan(sqlitedb,end_time=end_time,table_name_prefix=table_name_prefix) model_results3 = model_losss_yongan(sqlitedb,end_time=end_time,table_name_prefix=table_name_prefix)
# logger.info('训练数据绘图end') logger.info('训练数据绘图end')
# 模型报告 # 模型报告
logger.info('制作报告ing') logger.info('制作报告ing')

49
main_yuanyou.py
View File

@ -215,6 +215,9 @@ def predict_main():
warning_data_df.to_sql(table_name, con=engine, if_exists='append', index=False) warning_data_df.to_sql(table_name, con=engine, if_exists='append', index=False)
if is_update_warning_data: if is_update_warning_data:
upload_warning_info(len(warning_data_df)) upload_warning_info(len(warning_data_df))
# 发送钉钉消息
upload_warning_info(len(warning_data_df))
except: except:
logger.info('上传预警信息到数据库失败') logger.info('上传预警信息到数据库失败')
@ -226,32 +229,32 @@ 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(df,
horizon=horizon, # horizon=horizon,
input_size=input_size, # input_size=input_size,
train_steps=train_steps, # train_steps=train_steps,
val_check_steps=val_check_steps, # val_check_steps=val_check_steps,
early_stop_patience_steps=early_stop_patience_steps, # early_stop_patience_steps=early_stop_patience_steps,
is_debug=is_debug, # is_debug=is_debug,
dataset=dataset, # dataset=dataset,
is_train=is_train, # is_train=is_train,
is_fivemodels=is_fivemodels, # is_fivemodels=is_fivemodels,
val_size=val_size, # val_size=val_size,
test_size=test_size, # test_size=test_size,
settings=settings, # settings=settings,
now=now, # now=now,
etadata=etadata, # etadata=etadata,
modelsindex=modelsindex, # modelsindex=modelsindex,
data=data, # data=data,
is_eta=is_eta, # is_eta=is_eta,
end_time=end_time, # end_time=end_time,
) # )
logger.info('模型训练完成') logger.info('模型训练完成')
logger.info('训练数据绘图ing') logger.info('训练数据绘图ing')
model_results3 = model_losss(sqlitedb,end_time=end_time) # model_results3 = model_losss(sqlitedb,end_time=end_time)
logger.info('训练数据绘图end') logger.info('训练数据绘图end')
# 模型报告 # 模型报告
@ -259,8 +262,8 @@ def predict_main():
title = f'{settings}--{end_time}-预测报告' # 报告标题 title = f'{settings}--{end_time}-预测报告' # 报告标题
reportname = f'Brent原油大模型预测--{end_time}.pdf' # 报告文件名 reportname = f'Brent原油大模型预测--{end_time}.pdf' # 报告文件名
reportname = reportname.replace(':', '-') # 替换冒号 reportname = reportname.replace(':', '-') # 替换冒号
brent_export_pdf(dataset=dataset,num_models = 5 if is_fivemodels else 22,time=end_time, # brent_export_pdf(dataset=dataset,num_models = 5 if is_fivemodels else 22,time=end_time,
reportname=reportname,sqlitedb=sqlitedb), # reportname=reportname,sqlitedb=sqlitedb),
logger.info('制作报告end') logger.info('制作报告end')
logger.info('模型训练完成') logger.info('模型训练完成')