División en train-test
En este ejercicio vas a tomar el conjunto de datos de producción de caramelos y dividirlo en un conjunto de entrenamiento y otro de prueba. Como viste en el video, la razón para hacerlo es poder evaluar la calidad del ajuste de tu modelo cuando termines.
El conjunto de datos de producción de caramelos ya se ha cargado como candy y pyplot se ha importado como plt.
Este ejercicio forma parte del curso
Modelos ARIMA en Python
Instrucciones del ejercicio
- Divide la serie temporal en conjuntos de entrenamiento y prueba recortando con índices
datetime. Toma el conjunto de entrenamiento como todo hasta el final de 2006 y el de prueba como todo desde el inicio de 2007. - Crea unos ejes de
pyplotusando la funciónsubplots(). - Usa el método
.plot()del DataFrame para representar los conjuntos de entrenamiento y prueba en el ejeax.
ejercicio interactivo práctico
Prueba este ejercicio completando este código de ejemplo.
# 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()