|
|
import os
|
|
|
|
|
|
import pyecharts
|
|
|
from pyecharts.charts import Line
|
|
|
from pyecharts import options as opts
|
|
|
from pyecharts.globals import ThemeType
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
dir_path = r'E:\Data\Research\Outcome\Abt-Buy'
|
|
|
filename_list = os.listdir(dir_path)
|
|
|
iter_list = []
|
|
|
precision = []
|
|
|
recall = []
|
|
|
f1 = []
|
|
|
interpretability = []
|
|
|
performance = []
|
|
|
for _ in filename_list:
|
|
|
if _.startswith('eval_result'):
|
|
|
it = int(_[12:13])
|
|
|
iter_list.append(str(it))
|
|
|
with open(dir_path + '\\' + _, 'r') as f:
|
|
|
# 读取每一行的md，加入该文件的md列表
|
|
|
for line in f.readlines():
|
|
|
if line.startswith('Precision'):
|
|
|
lt = line.split(' ')
|
|
|
value = float(lt[2].replace('%', ''))/100
|
|
|
precision.append(value)
|
|
|
elif line.startswith('Recall'):
|
|
|
lt = line.split(' ')
|
|
|
value = float(lt[2].replace('%', ''))/100
|
|
|
recall.append(value)
|
|
|
elif line.startswith('F1'):
|
|
|
lt = line.split(' ')
|
|
|
value = float(lt[2].replace('%', ''))/100
|
|
|
f1.append(value)
|
|
|
elif line.startswith('interpretability'):
|
|
|
lt = line.split(':')
|
|
|
value = float(lt[1])
|
|
|
interpretability.append(value)
|
|
|
elif line.startswith('performance'):
|
|
|
lt = line.split(':')
|
|
|
value = float(lt[1])
|
|
|
performance.append(value)
|
|
|
|
|
|
line = (
|
|
|
Line(init_opts=opts.InitOpts(theme=ThemeType.LIGHT))
|
|
|
.add_xaxis(iter_list)
|
|
|
.add_yaxis('Precision', precision)
|
|
|
.add_yaxis('Recall', recall)
|
|
|
.add_yaxis('F1', f1)
|
|
|
.add_yaxis('Interpretability', interpretability)
|
|
|
.add_yaxis('Performance', performance)
|
|
|
.set_global_opts(title_opts=opts.TitleOpts(title=dir_path.split('\\')[-1]))
|
|
|
)
|
|
|
line.render(dir_path + '\\' + "line.html")
|
