Mulai sekarangMulai gratis

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 merupakan bagian dari kursus

Machine Learning dengan Model Berbasis Pohon di Python

Lihat Kursus

Instruksi latihan

  • Prediksi label himpunan uji menggunakan model regresi linear (lr) dan simpan hasilnya ke y_pred_lr.

  • Hitung MSE himpunan uji dan simpan hasilnya ke mse_lr.

  • Hitung RMSE himpunan uji dan simpan hasilnya ke rmse_lr.

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

# 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))
Edit dan Jalankan Kode