MulaiMulai sekarang secara gratis

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

Lihat Kursus

Petunjuk latihan

  • Impor mean_squared_error sebagai MSE dari sklearn.metrics.

  • Ekstrak estimator terbaik dari grid_rf dan simpan ke best_model.

  • Prediksi label himpunan uji best_model dan simpan hasilnya ke y_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)) 
Edit dan Jalankan Kode