我怎么知道每个分类功能的性能

时间:2018-03-12 04:12:36

标签: python

sunsList.add(
    new SunSession(
       "&#xf047"
 }

我在训练数据中使用逻辑回归建立模型我有三个特征,一个特征是分类特征(前颜色:蓝色,红色,绿色)。我转换成虚拟变量并构建模型。我想知道每种类型的性能绿色红色代替整个

1 个答案:

答案 0 :(得分:0)

假设存储分类数据的特征在第一列中,因此索引为0。您可以将X_test中与每个类别相关联的数据提取为

col = 0
for i in set(X_test[:,col]):
    ixs = X_test[:,col] == i
    print('Category ' + str(i) + ': ') 
    print(metrics.accuracy_score(X_test[ixs], y_test[ixs])) 

这将为您提供第1栏中每个类别的准确性。