Навчіть просту модель
Як ви визначили, це задача регресії. Тож ви готові побудувати модель для подальшої подачі. Але замість найпростішої лінійної регресії зі слайдів побудуємо готову з коробки модель Random Forest.
Ви використаєте клас RandomForestRegressor з бібліотеки scikit-learn.
Ваша мета — навчити модель Random Forest з типовими параметрами на ознаках «store» і «item».
Ця вправа є частиною курсу
Як перемагати в змаганнях Kaggle за допомогою Python
Інструкції до вправи
- Зчитайте тренувальні дані за допомогою
pandas. - Створіть об'єкт Random Forest.
- Навчіть модель Random Forest на ознаках «store» і «item» з цільовою змінною «sales».
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
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['____'])