Regresi linear vs pohon regresi
Pada latihan ini, Anda akan membandingkan RMSE himpunan uji dari dt dengan yang dicapai oleh model regresi linear. Kami telah membuat instance model regresi linear lr dan melatihnya pada himpunan data yang sama dengan dt.
Matriks fitur X_test, larik label y_test, model regresi linear terlatih lr, fungsi mean_squared_error yang diimpor dengan alias MSE, serta rmse_dt dari latihan sebelumnya tersedia di ruang kerja Anda.
Latihan ini adalah bagian dari kursus
Machine Learning dengan Model Berbasis Pohon di Python
Petunjuk latihan
Prediksi label himpunan uji menggunakan model regresi linear (
lr) dan simpan hasilnya key_pred_lr.Hitung MSE himpunan uji dan simpan hasilnya ke
mse_lr.Hitung RMSE himpunan uji dan simpan hasilnya ke
rmse_lr.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Predict test set labels
____ = ____.____(____)
# Compute mse_lr
____ = ____(____, ____)
# Compute rmse_lr
____ = ____
# Print rmse_lr
print('Linear Regression test set RMSE: {:.2f}'.format(rmse_lr))
# Print rmse_dt
print('Regression Tree test set RMSE: {:.2f}'.format(rmse_dt))