MulaiMulai sekarang secara gratis

Evaluasi galat pelatihan

Sekarang Anda akan mengevaluasi RMSE himpunan latih yang dicapai oleh pohon regresi dt yang telah Anda inisialisasi pada latihan sebelumnya.

Selain dt, X_train dan y_train tersedia di workspace Anda.

Perhatikan bahwa di scikit-learn, MSE sebuah model dapat dihitung sebagai berikut:

MSE_model = mean_squared_error(y_true, y_predicted)

di mana kita menggunakan fungsi mean_squared_error dari modul metrics dan memberikan label sebenarnya y_true sebagai argumen pertama, serta label prediksi dari model y_predicted sebagai argumen kedua.

Latihan ini adalah bagian dari kursus

Machine Learning dengan Model Berbasis Pohon di Python

Lihat Kursus

Petunjuk latihan

  • Impor mean_squared_error sebagai MSE dari sklearn.metrics.
  • Latih dt pada himpunan latih.
  • Prediksi label himpunan latih dengan dt dan simpan hasilnya ke y_pred_train.
  • Evaluasi RMSE himpunan latih dt dan simpan ke RMSE_train.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Import mean_squared_error from sklearn.metrics as MSE
____

# Fit dt to the training set
____.____(____, ____)

# Predict the labels of the training set
____ = ____.____(____)

# Evaluate the training set RMSE of dt
____ = (____(____, ____))**(___)

# Print RMSE_train
print('Train RMSE: {:.2f}'.format(RMSE_train))
Edit dan Jalankan Kode