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 2024-11-05 14:52:46 +08:00
parent a68ed336de
commit 2561958c58
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
models/nerulforcastmodels.py
View File

@ -631,7 +631,7 @@ def brent_export_pdf(num_indicators=475,num_models=21, num_dayindicator=202,inpu
df_zhibiaoshuju = data.copy() # 气泡图用
# 绘制特征相关气泡图
content.append(Graphs.draw_text('气泡图中，横轴为指标名称，纵轴为指标名称，面积越大表示相关性越大，面积越小表示相关性越小。'))
grouped = df_zhibiaofenlei.groupby('指标分类')
grouped_corr = pd.DataFrame(columns=['指标分类', '指标数量', '相关性总和'])
@ -705,6 +705,7 @@ def brent_export_pdf(num_indicators=475,num_models=21, num_dayindicator=202,inpu
plt.savefig(os.path.join(dataset, '指标分类相关性总和的气泡图.png'), bbox_inches='tight')
plt.close()
content.append(Graphs.draw_img(os.path.join(dataset,'指标分类相关性总和的气泡图.png')))
content.append(Graphs.draw_text('气泡图中，横轴为指标分类，纵轴为指标分类下的特征数量，气泡的面积越大表示该分类中特征的相关系数和越大。'))
logger.info(f'绘制相关性总和的气泡图结束')