比较 one-hot-encoded 列 header 和预测标签

Compare one-hot-encoded column header and predicted labels

我有 3 个 one-hot-encoded 列,其中 header 名称是标签,还有一个预测列 preds,其中预测标签(见图)。我想通过比较 preds 中的标签和 1 编码列 header.

来计算我的预测性能

在这个例子中,我只有 20% 预测正确。

在 pandas 中有快速的方法吗?

IIUC,DataFrame.lookup and np.mean

df[['Type_1','Type_2','Type_3']].lookup(df.index, df['preds']).mean() * 100