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())