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
Petunjuk latihan
- Impor
mean_squared_errorsebagaiMSEdarisklearn.metrics. - Latih
dtpada himpunan latih. - Prediksi label himpunan latih dengan
dtdan simpan hasilnya key_pred_train. - Evaluasi RMSE himpunan latih
dtdan simpan keRMSE_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))