Bắt đầu ngayBắt đầu miễn phí

Trực quan hóa mức độ quan trọng của đặc trưng

Trong bài tập này, bạn sẽ xác định những đặc trưng nào có tính dự đoán cao nhất theo bộ random forests regressor rf mà bạn đã huấn luyện ở bài trước.

Để làm điều đó, bạn sẽ vẽ một biểu đồ thanh ngang thể hiện mức độ quan trọng của đặc trưng do rf đánh giá. May mắn là việc này khá dễ nhờ khả năng vẽ của pandas.

Chúng tôi đã tạo một đối tượng pandas.Series tên là importances, trong đó chứa tên đặc trưng ở index và giá trị mức độ quan trọng tương ứng. Ngoài ra, matplotlib.pyplot đã được nạp sẵn dưới tên pltpandaspd.

Bài tập này là một phần của khóa học

Machine Learning với mô hình dựa trên cây trong Python

Xem khóa học

Hướng dẫn bài tập

  • Gọi phương thức .sort_values() trên importances và gán kết quả cho importances_sorted.

  • Gọi phương thức .plot() trên importances_sorted và thiết lập các tham số:

    • kind'barh'
    • color'lightgreen'

Bài tập tương tác thực hành trực tiếp

Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.

# Create a pd.Series of features importances
importances = pd.Series(data=rf.feature_importances_,
                        index= X_train.columns)

# Sort importances
importances_sorted = ____

# Draw a horizontal barplot of importances_sorted
____.____(____='____', ____='____')
plt.title('Features Importances')
plt.show()
Chỉnh sửa và Chạy Mã