mlprimitives.adapters.statsmodels module

class mlprimitives.adapters.statsmodels.ARIMA(p, d, q, steps)

Bases: object

A Wrapper for the statsmodels.tsa.arima_model.ARIMA class.

predict(X)

Predict values using the initialized object.

Parameters

X (ndarray) – N-dimensional array containing the input sequences for the model.

Returns

N-dimensional array containing the predictions for each input sequence.

Return type

ndarray