MulaiMulai sekarang secara gratis

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

Lihat Kursus

Petunjuk latihan

  • Impor fungsi mean_squared_error sebagai MSE dari sklearn.metrics.
  • Prediksikan label himpunan data uji dan simpan keluarannya ke y_pred.
  • Hitung MSE himpunan data uji dengan memanggil MSE dan simpan hasilnya ke mse_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))
Edit dan Jalankan Kode