Chia tập train-test
Trong bài này, bạn sẽ lấy bộ dữ liệu sản xuất kẹo và chia nó thành tập train và tập test. Như bạn đã hiểu trong video, mục đích là để bạn có thể kiểm tra chất lượng độ khớp của mô hình sau khi hoàn thành.
Bộ dữ liệu sản xuất kẹo đã được nạp sẵn dưới tên candy và pyplot đã được nạp dưới tên plt.
Bài tập này là một phần của khóa học
Mô hình ARIMA với Python
Hướng dẫn bài tập
- Chia chuỗi thời gian thành tập train và test bằng cách cắt theo các chỉ mục
datetime. Lấy tập train là mọi dữ liệu đến hết năm 2006 và tập test là mọi dữ liệu từ đầu năm 2007. - Tạo một axes của
pyplotbằng hàmsubplots(). - Dùng phương thức
.plot()của DataFrame để vẽ tập train và test trên trụcax.
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.
# Split the data into a train and test set
candy_train = candy.____
candy_test = candy.____
# Create an axis
fig, ax = ____
# Plot the train and test sets on the axis ax
candy_train.____(ax=____)
candy_test.____(ax=____)
plt.show()