1. Learn
  2. /
  3. Courses
  4. /
  5. ARIMA Models in Python

Exercise

Differencing and fitting ARMA

In this exercise you will fit an ARMA model to the Amazon stocks dataset. As you saw before, this is a non-stationary dataset. You will use differencing to make it stationary so that you can fit an ARMA model.

In the next section you'll make a forecast of the differences and use this to forecast the actual values.

The Amazon stock time series is available in your environment as amazon. The ARIMA model class is also available in your environment.

Instructions

100 XP
  • Use the .diff() method of amazon to make the time series stationary by taking the first difference. Don't forget to drop the NaN values using the .dropna() method.
  • Create an ARMA(2,2) model using the ARIMA class, passing it the stationary data.
  • Fit the model.