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

2
config_jingbo_pro.py
View File

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

6
main_yongan.py
View File

@ -242,9 +242,9 @@ def predict_main():
# logger.info('模型训练完成')
# logger.info('训练数据绘图ing')
# model_results3 = model_losss_yongan(sqlitedb,end_time=end_time,table_name_prefix=table_name_prefix)
# logger.info('训练数据绘图end')
logger.info('训练数据绘图ing')
model_results3 = model_losss_yongan(sqlitedb,end_time=end_time,table_name_prefix=table_name_prefix)
logger.info('训练数据绘图end')
# 模型报告
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)
if is_update_warning_data:
upload_warning_info(len(warning_data_df))
# 发送钉钉消息
upload_warning_info(len(warning_data_df))
except:
logger.info('上传预警信息到数据库失败')
@ -226,32 +229,32 @@ def predict_main():
row, col = df.shape
now = datetime.datetime.now().strftime('%Y%m%d%H%M%S')
ex_Model(df,
horizon=horizon,
input_size=input_size,
train_steps=train_steps,
val_check_steps=val_check_steps,
early_stop_patience_steps=early_stop_patience_steps,
is_debug=is_debug,
dataset=dataset,
is_train=is_train,
is_fivemodels=is_fivemodels,
val_size=val_size,
test_size=test_size,
settings=settings,
now=now,
etadata=etadata,
modelsindex=modelsindex,
data=data,
is_eta=is_eta,
end_time=end_time,
)
# ex_Model(df,
# horizon=horizon,
# input_size=input_size,
# train_steps=train_steps,
# val_check_steps=val_check_steps,
# early_stop_patience_steps=early_stop_patience_steps,
# is_debug=is_debug,
# dataset=dataset,
# is_train=is_train,
# is_fivemodels=is_fivemodels,
# val_size=val_size,
# test_size=test_size,
# settings=settings,
# now=now,
# etadata=etadata,
# modelsindex=modelsindex,
# data=data,
# is_eta=is_eta,
# end_time=end_time,
# )
logger.info('模型训练完成')
logger.info('训练数据绘图ing')
model_results3 = model_losss(sqlitedb,end_time=end_time)
# model_results3 = model_losss(sqlitedb,end_time=end_time)
logger.info('训练数据绘图end')
# 模型报告
@ -259,8 +262,8 @@ def predict_main():
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,sqlitedb=sqlitedb),
# brent_export_pdf(dataset=dataset,num_models = 5 if is_fivemodels else 22,time=end_time,
# reportname=reportname,sqlitedb=sqlitedb),
logger.info('制作报告end')
logger.info('模型训练完成')