diff --git a/visualization.py b/visualization.py deleted file mode 100644 index 4740d1d..0000000 --- a/visualization.py +++ /dev/null @@ -1,24 +0,0 @@ -import matplotlib.pyplot as plt - -# 设置中文字体 -plt.rcParams["font.family"] = ["SimHei"] - -def plot_views_distribution(main_views, filename="用户观点分布.png"): - """绘制用户观点分布柱状图""" - plt.figure(figsize=(10, 6)) - plt.bar( - main_views.keys(), - main_views.values(), - color=['#4CAF50', '#2196F3', '#f44336', '#FFC107', '#9E9E9E'] - ) - plt.title('用户对大语言模型的主要关注点分布') - plt.ylabel('弹幕数量') - plt.xticks(rotation=30) - - # 添加数值标签 - for i, v in enumerate(main_views.values()): - plt.text(i, v + 5, str(v), ha='center') - - plt.tight_layout() - plt.savefig(filename, dpi=300) - plt.show() \ No newline at end of file