我有一个数据框,如下所示.
id label prediction
1 cat cat
2 dog cat
3 cow dog
4 cow cow
5 dog cat
6 cat cat
7 cat cat
8 dog dog
9 dog dog
10 cat cat
根据上述df,我想用Pandas 计算整体精度.
我try 了下面的代码来计算类精度.
class_wise_accuracy = (df.groupby('label')['prediction']
.value_counts(normalize=True)
.unstack(fill_value=0)
)
confusion_matrix = (df.groupby('label')['prediction']
.value_counts()
.unstack(fill_value=0)
.reset_index()
)
预期输出:
overall_accuracy = (4+1+2)/df.shape[0] = 0.7