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 plt và pandas là pd.
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
Hướng dẫn bài tập
Gọi phương thức
.sort_values()trênimportancesvà gán kết quả choimportances_sorted.Gọi phương thức
.plot()trênimportances_sortedvà thiết lập các tham số:kindlà'barh'colorlà'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()