Python 分类模型:如何检查每个分类的分数
我正在使用RandomForestClassifier将数据分类为两种类型——0或1。目前我正在使用下面的代码,并获得所有测试数据的总分。我想做的是得到类型0数据和类型1数据的单独分数。感谢您的帮助Python 分类模型:如何检查每个分类的分数,python,pandas,numpy,scikit-learn,Python,Pandas,Numpy,Scikit Learn,我正在使用RandomForestClassifier将数据分类为两种类型——0或1。目前我正在使用下面的代码,并获得所有测试数据的总分。我想做的是得到类型0数据和类型1数据的单独分数。感谢您的帮助 X = features_enc Y = np.asarray(df[target_column]) x_train, x_test, y_train, y_test = train_test_split(X, Y, test_size=0.33, random_state=42) rando
X = features_enc
Y = np.asarray(df[target_column])
x_train, x_test, y_train, y_test = train_test_split(X, Y, test_size=0.33, random_state=42)
random.seed(100)
rf = RandomForestClassifier(n_estimators=100)
rf.fit(x_train, y_train)
score = rf.score(x_test, y_test)
print(score)
您可以从sklearn导入以下内容:
来自sklearn.metrics导入分类报告
这将为您提供所有可能的分数:
#--snip--
预测=射频预测(x_测试)
打印(分类报告(y检验,预测))
这将打印格式良好的评估。有关更多信息,请参阅