sklearn【F1 Scoree】F1分数原理及实战代码！_precision和recall都很高

一、F1 Scoree 介绍

在分类任务中，评估模型的性能是至关重要的。除了准确率（Accuracy）之外，我们还需要考虑其他指标，如精确度（Precision）和查全率（Recall），以更全面地了解模型在识别不同类别样本时的表现。而F1分数（F1 Score）是Precision和Recall的调和平均值，能够综合这两个指标，为我们提供一个更加平衡的评估模型性能的视角。

本文将介绍如何使用sklearn库来计算F1分数，并解释其背后的计算原理。

首先，我们需要理解F1分数的计算方式。F1分数综合考虑了Precision和Recall，其计算公式为：

$$F1Score=2(Precision\ast Recall)/(Precision+Recall)$$

其中：

• Precision 衡量的是模型预测为正例的样本中真正为正例的比例。
• Recall 衡量的是所有真正为正例的样本中被模型正确预测为正例的比例。

二、案例学习

sklearn库中提供了f1_score函数，可以方便地计算F1分数。下面我们将通过一个简单的例子来演示如何使用这个函数。

首先，我们需要导入必要的库，并准备数据集。在这个例子中，我们将使用sklearn自带的鸢尾花（Iris）数据集，并将其转化为二分类问题来简化演示。

from sklearn import datasets
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import label_binarize
from sklearn.metrics import f1_score
from sklearn.linear_model import LogisticRegression

# 加载鸢尾花数据集
iris = datasets.load_iris()
X = iris.data
y = iris.target

# 将多分类问题转化为二分类问题（例如，只区分山鸢尾和非山鸢尾）
y = label_binarize(y, classes=[2])
y = y.ravel()

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

接下来，我们创建一个分类模型，并使用训练数据进行训练。在这个例子中，我们将使用逻辑回归作为分类器。

# 创建逻辑回归模型
model = LogisticRegression()

# 使用训练数据进行训练
model.fit(X_train, y_train)
1
2
3
4
5

训练完成后，我们可以使用模型对测试集进行预测，并使用sklearn的f1_score函数来计算F1分数。

# 对测试集进行预测
y_pred = model.predict(X_test)

# 计算F1分数
f1 = f1_score(y_test, y_pred)
print(f'F1 Score: {f1:.2f}')

# F1 Score: 0.99
1
2
3
4
5
6
7
8

在上面的代码中，f1_score函数接受实际的目标值（y_test）和模型预测的值（y_pred）作为输入，然后返回F1分数的值。F1分数是一个介于0和1之间的值，越接近1表示模型的性能越好。

需要注意的是，F1分数综合考虑了Precision和Recall，因此它能够更全面地评估模型在识别正例和反例时的性能。当Precision和Recall都很高时，F1分数也会很高，这表示模型在识别正例和反例时都表现良好。相反，如果Precision和Recall中有一个较低，F1分数也会受到影响，这可以帮助我们发现模型在哪方面存在问题，从而进行针对性的优化。

此外，当数据集的正反例样本数不平衡时，F1分数特别有用。准确率可能会因为多数类的正确分类而被高估，而F1分数则能够更准确地反映模型在少数类上的性能。

三、总结

通过上面的示例，我们展示了如何使用sklearn来训练和评估一个分类模型，并计算其在测试集上的F1分数。在实际应用中，我们可以根据具体的问题和数据集选择合适的分类模型和评估指标，以得到更准确的分类结果和性能评估。F1分数作为一个综合指标，能够帮助我们更全面地了解模型的性能，并指导我们进行模型的优化和改进。