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
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_testdany_pred. - Cetak
r_squareddanrmse.
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(____))