Träna en enkel modell
Som du konstaterat handlar det här om ett regressionsproblem. Nu är du redo att bygga en modell inför en kommande inlämning. I stället för att bygga den enklaste linjära regressionsmodellen – som i genomgången – ska du nu bygga en Random Forest-modell med standardinställningar.
Du kommer att använda klassen RandomForestRegressor från biblioteket scikit-learn.
Målet är att träna en Random Forest-modell med standardparametrar på särdragen "store" och "item".
Den här övningen är en del av kursen
Vinna en Kaggle-tävling i Python
Övningsinstruktioner
- Läs in träningsdatan med
pandas. - Skapa ett Random Forest-objekt.
- Träna Random Forest-modellen på särdragen "store" och "item" med "sales" som målvariabel.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
import pandas as pd
from sklearn.ensemble import RandomForestRegressor
# Read the train data
train = ____.____('train.csv')
# Create a Random Forest object
rf = ____()
# Train a model
rf.fit(X=train[['store', ____]], y=train['____'])