MulaiMulai sekarang secara gratis

Kinerja regresi

Sekarang Anda telah melatih sebuah model, reg, menggunakan semua fitur dari sales_df, dan membuat prediksi nilai penjualan, Anda dapat mengevaluasi kinerjanya menggunakan beberapa metrik regresi umum.

Variabel X_train, X_test, y_train, y_test, dan y_pred, beserta model yang telah dipasang, reg, semuanya dari latihan sebelumnya, telah dimuatkan untuk Anda.

Tugas Anda adalah mengetahui seberapa baik fitur dapat menjelaskan varians pada nilai target, sekaligus menilai kemampuan model membuat prediksi pada data yang belum terlihat.

Latihan ini adalah bagian dari kursus

Supervised Learning dengan scikit-learn

Lihat Kursus

Petunjuk latihan

  • Import root_mean_squared_error.
  • Hitung skor R-squared model dengan meneruskan nilai fitur uji dan nilai target uji ke metode yang sesuai.
  • Hitung root mean squared error model menggunakan y_test dan y_pred.
  • Cetak r_squared dan rmse.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Import root_mean_squared_error
from ____.____ import ____

# Compute R-squared
r_squared = reg.____(____, ____)

# Compute RMSE
rmse = ____(____, ____)

# Print the metrics
print("R^2: {}".format(____))
print("RMSE: {}".format(____))
Edit dan Jalankan Kode