Predicții pe setul de testare

În exercițiul anterior, regresia liniară și ridge au produs rezultate similare. Oricare dintre aceste modele ar fi o alegere potrivită; totuși, poți verifica performanța predictivă pe setul de testare pentru a vedea dacă vreunul îl depășește pe celălalt.

Vei folosi eroarea pătratică medie de rădăcină (RMSE) ca metrică. Dicționarul models, care conține numele și instanțele celor două modele, a fost preîncărcat pentru tine, împreună cu tablourile de antrenament și țintă X_train_scaled, X_test_scaled, y_train și y_test.

Importă root_mean_squared_error.
Antrenează modelul pe caracteristicile de antrenament scalate și etichetele de antrenament.
Realizează predicții folosind caracteristicile de testare scalate.
Calculează RMSE pasând etichetele din setul de testare și etichetele prezise.

.css-6su6fj{-webkit-flex-shrink:0;-ms-flex-negative:0;flex-shrink:0;}exercițiu

Instrucțiuni

exercițiu