Evaluasi hutan optimal
Pada latihan terakhir di kursus ini, Anda akan mengevaluasi RMSE himpunan uji dari model optimal grid_rf.
Himpunan data sudah dimuat dan diproses untuk Anda serta dibagi menjadi 80% latih dan 20% uji. Di lingkungan Anda tersedia X_test, y_test, dan fungsi mean_squared_error dari sklearn.metrics dengan alias MSE. Selain itu, kami juga telah memuat objek GridSearchCV terlatih grid_rf yang Anda buat pada latihan sebelumnya. Perhatikan bahwa grid_rf telah dilatih sebagai berikut:
grid_rf.fit(X_train, y_train)
Latihan ini adalah bagian dari kursus
Machine Learning dengan Model Berbasis Pohon di Python
Petunjuk latihan
Impor
mean_squared_errorsebagaiMSEdarisklearn.metrics.Ekstrak estimator terbaik dari
grid_rfdan simpan kebest_model.Prediksi label himpunan uji
best_modeldan simpan hasilnya key_pred.Hitung RMSE himpunan uji
best_model.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Import mean_squared_error from sklearn.metrics as MSE
____
# Extract the best estimator
best_model = ____
# Predict test set labels
y_pred = ____
# Compute rmse_test
rmse_test = ____
# Print rmse_test
print('Test RMSE of best model: {:.3f}'.format(rmse_test))