我如何知道每个分类特征的性能

Question

from sklearn.cross_validation import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=0)

from sklearn.linear_model import LogisticRegression
lr.fit(X_train, y_train)

print(metrics.accuracy_score(y_test, y_pred)) 

我在训练数据中使用逻辑回归构建模型我有三个特征一个特征是分类特征（例如颜色：蓝色、红色、绿色）。我转换为虚拟变量并构建模型。我想知道每种类型的表现都是绿色红色蓝色而不是整体

Answer 1

假设存储分类数据的特征在第一列，因此索引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])) 

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