开始使用免费开始使用

F1 分数

正如您已经发现的,精确率与召回率之间存在权衡。两者都很重要,且根据业务建模流失(churn)的方式,您可能需要优先优化其中一个。很多情况下,相关方希望有一个能够量化模型表现的单一指标。AUC 是一种可用的指标,另一种是 F1 分数,其计算方式如下:

2 * (precision * recall) / (precision + recall)

F1 分数的优势在于它把精确率与召回率融合为单一指标。在类别不平衡的情况下,只要 F1 分数高,也说明模型表现良好。在 scikit-learn 中,您可以使用 f1_score 函数来计算 F1 分数。

本练习是课程的一部分

营销分析:用 Python 预测客户流失

查看课程

练习说明

  • sklearn.metrics 导入 f1_score
  • 输出已训练随机森林的 F1 分数。

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Instantiate the classifier
clf = RandomForestClassifier()

# Fit to the training data
clf.fit(X_train, y_train)

# Predict the labels of the test set
y_pred = clf.predict(X_test)

# Import f1_score


# Print the F1 score
print(____)
编辑并运行代码