開始使用免費開始

Entropy 與 Gini 指標比較

在這個練習中,你要比較 dt_entropy 的測試集正確率,和另一棵名為 dt_gini 的樹的正確率。dt_gini 使用相同的資料集與參數訓練,但資訊準則改為使用關鍵字 'gini' 設定的 Gini 指標。

X_testy_testdt_entropy,以及代表 dt_gini 在測試集上所達成正確率的 accuracy_gini 都已提供在你的工作環境中。

本練習屬於課程

Machine Learning with Tree-Based Models in Python

檢視課程

練習說明

  • sklearn.metrics 匯入 accuracy_score
  • dt_entropy 的測試集做標籤預測,將結果指定給 y_pred
  • 評估 dt_entropy 在測試集上的正確率,將結果指定給 accuracy_entropy
  • 檢視 accuracy_entropyaccuracy_gini

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# Import accuracy_score from sklearn.metrics
from ____.____ import ____

# Use dt_entropy to predict test set labels
____= ____.____(____)

# Evaluate accuracy_entropy
accuracy_entropy = ____(____, ____)

# Print accuracy_entropy
print(f'Accuracy achieved by using entropy: {accuracy_entropy:.3f}')

# Print accuracy_gini
print(f'Accuracy achieved by using the gini index: {accuracy_gini:.3f}')
編輯並執行程式碼