【发布时间】:2021-10-13 22:06:00
【问题描述】:
我正在尝试寻找accuarcy/PRecision/Reacll 等... 所以我使用了这段代码,它对我来说效果很好,但实际上我想将输出形式更改为表格 我的输出:
Column 2 acc: 1.0
Column 2 p: 1.0
Column 2 r: 1.0
Column 1 acc: 1.0
Column 1 p: 1.0
Column 1 r: 1.0
Column 3 acc: 1.0
Column 3 p: 1.0
Column 3 r: 1.0
我想要的输出:
+----------+-----------+-------+---------+
| Feature | Precision |Recall | Accuracy|
+----------+-----------+-------+---------+
| 1 | 1.0 | 1.0 | 1.0 |
| 2 | 1.0 | 1.0 | 1.0 |
| 3 | 1.0 | 1.0 | 1.0 |
+----------+----------+--------+---------+
我的代码:
def calc_acc(original, predect1):
common_columns = list(set(original.columns).intersection(predect1.columns))
avg_a = 0.0
avg_p = 0.0
avg_r = 0.0
for c in common_columns:
c_acc = accuracy_score(original[c], predect1[c])
p = precision_score(original[c], predect1[c], average='macro', labels=np.unique(predect1[c]))
r = recall_score(original[c], predect1[c], average='macro', labels=np.unique(predect1[c]))
print(f'Column {c} acc: {c_acc}')
print(f'Column {c} p: {p}')
print(f'Column {c} r: {r}')
avg_a += c_acc/len(common_columns)
avg_p += p/len(common_columns)
avg_r += r/len(common_columns)
NB:c 是列
【问题讨论】:
-
如何存储输出的数据?您还没有真正向任何人展示足够的代码来帮助您。请详细说明
-
@DarkKnight 我编辑了我的帖子
标签: python machine-learning scikit-learn prettytable tabular-form