1. 学ぶ
    2. /
  2. コース
    3. /
  3. Machine Learning với mô hình dựa trên cây trong Python
Connected

演習

Đánh giá lỗi huấn luyện

Giờ bạn sẽ đánh giá RMSE trên tập huấn luyện đạt được bởi cây hồi quy dt mà bạn đã khởi tạo ở bài trước.

Ngoài dt, X_train và y_train cũng có sẵn trong không gian làm việc của bạn.

Lưu ý: trong scikit-learn, MSE của một mô hình có thể được tính như sau:

MSE_model = mean_squared_error(y_true, y_predicted)

trong đó ta dùng hàm mean_squared_error từ module metrics và truyền vào nhãn thực y_true làm đối số thứ nhất, và nhãn dự đoán từ mô hình y_predicted làm đối số thứ hai.

指示

100 XP
  • Import mean_squared_error dưới tên MSE từ sklearn.metrics.
  • Fit dt trên tập huấn luyện.
  • Dự đoán nhãn trên tập huấn luyện bằng dt và gán kết quả cho y_pred_train.
  • Tính RMSE trên tập huấn luyện của dt và gán cho RMSE_train.