Huấn luyện một mô hình đơn giản
Như bạn đã xác định, đây là một bài toán hồi quy. Vậy giờ bạn đã sẵn sàng xây dựng mô hình để nộp bài. Nhưng thay vì xây dựng mô hình Linear Regression đơn giản như trong các slide, hãy thử một mô hình Random Forest mặc định (out-of-box).
Bạn sẽ dùng lớp RandomForestRegressor từ thư viện scikit-learn.
Mục tiêu của bạn là huấn luyện một mô hình Random Forest với tham số mặc định trên các đặc trưng "store" và "item".
Bài tập này là một phần của khóa học
Chinh phục cuộc thi Kaggle bằng Python
Hướng dẫn bài tập
- Đọc dữ liệu train bằng
pandas. - Tạo một đối tượng Random Forest.
- Huấn luyện mô hình Random Forest trên các đặc trưng "store" và "item" với "sales" là biến mục tiêu.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
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['____'])