Feature Selection for Time Series Forecasting with Python
The use of machine learning methods on time series data requires feature engineering.
A univariate time series dataset is only comprised of a sequence of observations. These must be transformed into input and output features in order to use supervised learning algorithms.
The problem is that there is little limit to the type and number of features you can engineer for a time series problem. Classical time series analysis tools like the correlogram can help with evaluating lag variables, but do not directly help when selecting other types of features, such as those derived from the timestamps (year, month or day) and moving statistics, like a moving average.
In this tutorial, you will discover how you can use the machine learning tools of feature importance and feature selection when working with time series data.
After completing this tutorial, you will know:
- How to create and interpret a correlogram of lagged observations.
- How to calculate and interpret feature importance scores for time series features.
- How to perform feature selection on time series input variables.
Let’s get started.
Tutorial Overview
This tutorial is broken down into the following 5 steps:
- Monthly Car Sales Dataset: That describes the dataset we will be working with.
- Make Stationary: That describes how to make the dataset stationary for analysis and forecasting.
- Autocorrelation Plot: That describes how to create a correlogram of the time series data.
- Feature Importance of Lag Variables: That describes how to calculate and review feature importance scores for time series data.
- Feature Selection of Lag Variables: That describes how to calculate and review feature selection results for time series data.
Let’s start off by looking at a standard time series dataset.
Stop learning Time Series Forecasting the slow way!
Take my free 7-day email course and discover data prep, modeling and more (with sample code).
Click to sign-up and also get a free PDF Ebook version of the course.
Monthly Car Sales Dataset
In this tutorial, we will use the Monthly Car Sales dataset.
This dataset describes the number of car sales in Quebec, Canada between 1960 and 1968.
The units are a count of the number of sales and there are 108 observations. The source data is credited to Abraham and Ledolter (1983).
Download the dataset and save it into your current working directory with the filename “car-sales.csv“. Note, you may need to delete the footer information from the file.
The code below loads the dataset as a Pandas Series object.
12345678910 | # line plot of time seriesfrom pandas import Seriesfrom matplotlib import pyplot# load datasetseries=Series.from_csv('car-sales.csv',header=0)# display first few rowsprint(series.head(5))# line plot of datasetseries.plot()pyplot.show() |
Running the example prints the first 5 rows of data.
1234567 | Month1960-01-01 65501960-02-01 87281960-03-01 120261960-04-01 143951960-05-01 14587Name: Sales, dtype: int64 |
A line plot of the data is also provided.
Make Stationary
We can see a clear seasonality and increasing trend in the data.
The trend and seasonality are fixed components that can be added to any prediction we make. They are useful, but need to be removed in order to explore any other systematic signals that can help make predictions.
A time series with seasonality and trend removed is called stationary.
To remove the seasonality, we can take the seasonal difference, resulting in a so-called seasonally adjusted time series.
The period of the seasonality appears to be one year (12 months). The code below calculates the seasonally adjusted time series and saves it to the file “seasonally-adjusted.csv“.
1234567891011121314 | # seasonally adjust the time seriesfrom pandas import Seriesfrom matplotlib import pyplot# load datasetseries=Series.from_csv('car-sales.csv',header=0)# seasonal differencedifferenced=series.diff(12)# trim off the first year of empty datadifferenced=differenced[12:]# save differenced dataset to filedifferenced.to_csv('seasonally_adjusted.csv')# plot differenced datasetdifferenced.plot()pyplot.show() |
Because the first 12 months of data have no prior data to be differenced against, they must be discarded.
The stationary data is stored in “seasonally-adjusted.csv“. A line plot of the differenced data is created.
The plot suggests that the seasonality and trend information was removed by differencing.
Autocorrelation Plot
Traditionally, time series features are selected based on their correlation with the output variable.
This is called autocorrelation and involves plotting autocorrelation plots, also called a correlogram. These show the correlation of each lagged observation and whether or not the correlation is statistically significant.
For example, the code below plots the correlogram for all lag variables in the Monthly Car Sales dataset.
123456 | from pandas import Seriesfrom statsmodels.graphics.tsaplots import plot_acffrom matplotlib import pyplotseries=Series.from_csv('seasonally_adjusted.csv',header=None)plot_acf(series)pyplot.show() |
Running the example creates a correlogram, or Autocorrelation Function (ACF) plot, of the data.
The plot shows lag values along the x-axis and correlation on the y-axis between -1 and 1 for negatively and positively correlated lags respectively.
The dots above the blue area indicate statistical significance. The correlation of 1 for the lag value of 0 indicates 100% positive correlation of an observation with itself.
The plot shows significant lag values at 1, 2, 12, and 17 months.
This analysis provides a good baseline for comparison.
Time Series to Supervised Learning
We can convert the univariate Monthly Car Sales dataset into a supervised learning problem by taking the lag observation (e.g. t-1) as inputs and using the current observation (t) as the output variable.
We can do this in Pandas using the shift function to create new columns of shifted observations.
The example below creates a new time series with 12 months of lag values to predict the current observation.
The shift of 12 months means that the first 12 rows of data are unusable as they contain NaN values.
12345678910111213 | from pandas import Seriesfrom pandas import DataFrame# load datasetseries=Series.from_csv('seasonally_adjusted.csv',header=None)# reframe as supervised learningdataframe=DataFrame()foriinrange(12,0,-1):dataframe['t-'+str(i)]=series.shift(i)dataframe['t']=series.valuesprint(dataframe.head(13))dataframe=dataframe[13:]# save to new filedataframe.to_csv('lags_12months_features.csv',index=False) |
Running the example prints the first 13 rows of data showing the unusable first 12 rows and the usable 13th row.
1234567891011121314151617181920212223242526272829 | t-12 t-11 t-10 t-9 t-8 t-7 t-6 t-5 \1961-01-01 NaN NaN NaN NaN NaN NaN NaN NaN1961-02-01 NaN NaN NaN NaN NaN NaN NaN NaN1961-03-01 NaN NaN NaN NaN NaN NaN NaN NaN1961-04-01 NaN NaN NaN NaN NaN NaN NaN NaN1961-05-01 NaN NaN NaN NaN NaN NaN NaN NaN1961-06-01 NaN NaN NaN NaN NaN NaN NaN 687.01961-07-01 NaN NaN NaN NaN NaN NaN 687.0 646.01961-08-01 NaN NaN NaN NaN NaN 687.0 646.0 -189.01961-09-01 NaN NaN NaN NaN 687.0 646.0 -189.0 -611.01961-10-01 NaN NaN NaN 687.0 646.0 -189.0 -611.0 1339.01961-11-01 NaN NaN 687.0 646.0 -189.0 -611.0 1339.0 30.01961-12-01 NaN 687.0 646.0 -189.0 -611.0 1339.0 30.0 1645.01962-01-01 687.0 646.0 -189.0 -611.0 1339.0 30.0 1645.0 -276.0 t-4 t-3 t-2 t-1 t1961-01-01 NaN NaN NaN NaN 687.01961-02-01 NaN NaN NaN 687.0 646.01961-03-01 NaN NaN 687.0 646.0 -189.01961-04-01 NaN 687.0 646.0 -189.0 -611.01961-05-01 687.0 646.0 -189.0 -611.0 1339.01961-06-01 646.0 -189.0 -611.0 1339.0 30.01961-07-01 -189.0 -611.0 1339.0 30.0 1645.01961-08-01 -611.0 1339.0 30.0 1645.0 -276.01961-09-01 1339.0 30.0 1645.0 -276.0 561.01961-10-01 30.0 1645.0 -276.0 561.0 470.01961-11-01 1645.0 -276.0 561.0 470.0 3395.01961-12-01 -276.0 561.0 470.0 3395.0 360.01962-01-01 561.0 470.0 3395.0 360.0 3440.0 |
The first 12 rows are removed from the new dataset and results are saved in the file “lags_12months_features.csv“.
This process can be repeated with an arbitrary number of time steps, such as 6 months or 24 months, and I would recommend experimenting.
Feature Importance of Lag Variables
Ensembles of decision trees, like bagged trees, random forest, and extra trees, can be used to calculate a feature importance score.
This is common in machine learning to estimate the relative usefulness of input features when developing predictive models.
We can use feature importance to help to estimate the relative importance of contrived input features for time series forecasting.
This is important because we can contrive not only the lag observation features above, but also features based on the timestamp of observations, rolling statistics, and much more. Feature importance is one method to help sort out what might be more useful in when modeling.
The example below loads the supervised learning view of the dataset created in the previous section, fits a random forest model (RandomForestRegressor), and summarizes the relative feature importance scores for each of the 12 lag observations.
A large-ish number of trees is used to ensure the scores are somewhat stable. Additionally, the random number seed is initialized to ensure that the same result is achieved each time the code is run.
1234567891011121314151617181920 | from pandas import read_csvfrom sklearn.ensemble import RandomForestRegressorfrom matplotlib import pyplot# load datadataframe=read_csv('lags_12months_features.csv',header=0)array=dataframe.values# split into input and outputX=array[:,0:-1]y=array[:,-1]# fit random forest modelmodel=RandomForestRegressor(n_estimators=500,random_state=1)model.fit(X,y)# show importance scoresprint(model.feature_importances_)# plot importance scoresnames=dataframe.columns.values[0:-1]ticks=[iforiinrange(len(names))]pyplot.bar(ticks,model.feature_importances_)pyplot.xticks(ticks,names)pyplot.show() |
Running the example first prints the importance scores of the lagged observations.
12 | [ 0.21642244 0.06271259 0.05662302 0.05543768 0.07155573 0.08478599 0.07699371 0.05366735 0.1033234 0.04897883 0.1066669 0.06283236] |
The scores are then plotted as a bar graph.
The plot shows the high relative importance of the observation at t-12 and, to a lesser degree, the importance of observations at t-2 and t-4.
It is interesting to note a difference with the outcome from the correlogram above.
This process can be repeated with different methods that can calculate importance scores, such as gradient boosting, extra trees, and bagged decision trees.
Feature Selection of Lag Variables
We can also use feature selection to automatically identify and select those input features that are most predictive.
A popular method for feature selection is called Recursive Feature Selection (RFE).
RFE works by creating predictive models, weighting features, and pruning those with the smallest weights, then repeating the process until a desired number of features are left.
The example below uses RFE with a random forest predictive model and sets the desired number of input features to 4.
12345678910111213141516171819202122232425 | from pandas import read_csvfrom sklearn.feature_selection import RFEfrom sklearn.ensemble import RandomForestRegressorfrom matplotlib import pyplot# load datasetdataframe=read_csv('lags_12months_features.csv',header=0)# separate into input and output variablesarray=dataframe.valuesX=array[:,0:-1]y=array[:,-1]# perform feature selectionrfe=RFE(RandomForestRegressor(n_estimators=500,random_state |