1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Học có giám sát với scikit-learn
Connected

Bài tập

Dự đoán trên tập kiểm tra

Trong bài trước, linear regression và ridge cho kết quả khá tương đồng. Bạn có thể chọn một trong hai mô hình; tuy nhiên, hãy kiểm tra hiệu năng dự đoán trên tập kiểm tra để xem liệu có mô hình nào vượt trội hơn không.

Bạn sẽ dùng chỉ số root mean squared error (RMSE). Từ điển models, chứa tên và instance của hai mô hình, đã được nạp sẵn cùng với các mảng đặc trưng và nhãn X_train_scaled, X_test_scaled, y_train, và y_test.

Hướng dẫn

100 XP
  • Import root_mean_squared_error.
  • Huấn luyện mô hình với các đặc trưng huấn luyện đã scale và nhãn huấn luyện.
  • Tạo dự đoán sử dụng các đặc trưng kiểm tra đã scale.
  • Tính RMSE bằng cách truyền các nhãn của tập kiểm tra và các nhãn dự đoán.