Evaluasi pohon regresi
Dalam latihan ini, Anda akan mengevaluasi kinerja himpunan data uji dari dt menggunakan metrik Root Mean Squared Error (RMSE). RMSE sebuah model mengukur, rata-rata, seberapa besar perbedaan prediksi model dibandingkan label sebenarnya. RMSE sebuah model dapat diperoleh dengan menghitung akar kuadrat dari Mean Squared Error (MSE) model.
Matriks fitur X_test, array y_test, serta decision tree regressor dt yang Anda latih pada latihan sebelumnya tersedia di workspace Anda.
Latihan ini adalah bagian dari kursus
Machine Learning dengan Model Berbasis Pohon di Python
Petunjuk latihan
- Impor fungsi
mean_squared_errorsebagaiMSEdarisklearn.metrics. - Prediksikan label himpunan data uji dan simpan keluarannya ke
y_pred. - Hitung MSE himpunan data uji dengan memanggil
MSEdan simpan hasilnya kemse_dt. - Hitung RMSE himpunan data uji dan simpan ke
rmse_dt.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Import mean_squared_error from sklearn.metrics as MSE
from ____.____ import ____ as ____
# Compute y_pred
____ = ____.____(____)
# Compute mse_dt
____ = ____(____, ____)
# Compute rmse_dt
____ = ____
# Print rmse_dt
print("Test set RMSE of dt: {:.2f}".format(rmse_dt))