Uitdaging: Prijsvoorspellingen Met Behulp van Polynomiale Regressie
Voor deze uitdaging bouw je dezelfde polynomiale regressie van graad 2 als in de vorige uitdaging. Je moet echter de dataset splitsen in een trainingsset en een testset om de RMSE voor beide sets te berekenen. Dit is nodig om te beoordelen of het model overfit of underfit.
Hier volgt een herinnering aan de train_test_split()
functie die je wilt gebruiken.
En ook een herinnering aan de mean_squared_error()
functie die nodig is om de RMSE te berekenen:
rmse = mean_squared_error(y_true, y_predicted, squared=False)
Swipe to start coding
- Wijs de DataFrame met één kolom
'age'
vandf
toe aan de variabeleX
. - Preprocess de
X
met behulp van de klassePolynomialFeatures
. - Splits de dataset met de juiste functie uit
sklearn
. - Bouw en train een model op de trainingsset.
- Voorspel de targets van zowel de trainings- als de testset.
- Bereken de RMSE voor zowel de trainings- als de testset.
- Print de samenvattingstabel.
Oplossing
Wanneer je de taak voltooit, zul je merken dat de test-RMSE zelfs lager is dan de trainings-RMSE. Gewoonlijk laten modellen geen betere resultaten zien op onbekende gegevens. Hier is het verschil klein en veroorzaakt door toeval. Onze dataset is relatief klein en bij het splitsen heeft de testset iets betere (makkelijker te voorspellen) datapunten gekregen.
Bedankt voor je feedback!