minimal_example.py 448 Bytes
Newer Older
chenzk's avatar
v1.0  
chenzk committed
1
2
3
4
5
6
7
8
9
10
11
12
13
from neuralforecast import NeuralForecast
from neuralforecast.models import iTransformer
from neuralforecast.utils import AirPassengersDF

horizon =8
nf = NeuralForecast(
    # models = [iTransformer( h=12, input_size=24, n_series=1, max_steps=100)],
    models = [iTransformer(h=horizon, input_size=2*horizon, n_series=4, max_steps=1000, early_stop_patience_steps=3)],
    freq = 'M'
)

nf.fit(df=AirPassengersDF, val_size=20)
print(nf.predict())