開始使用免費開始

F1 分數

你已經發現 precision 和 recall 之間存在權衡。兩者都是重要的評估指標,而你可能會依據企業想要如何建構流失預測模型,選擇著重最佳化其中一個。利害關係人往往希望有單一指標能量化模型效能。AUC 是其中一種可用的指標,另一種則是 F1 分數,其計算方式如下:

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

F1 分數的優點是把 precision 和 recall 合併為單一指標;即使在類別不平衡的情況下,F1 分數高通常也代表模型表現良好。在 scikit-learn 中,你可以使用 f1_score 函式計算 f-1 分數。

本練習屬於課程

行銷分析:用 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(____)
編輯並執行程式碼