AUNET is a neural network architecture designed for time series forecasting, combining multi-head self-attention with dense layers to capture temporal patterns in numeric datasets. Developed using TensorFlow/Keras, it supports customizable input windows and forecast horizons.
1from aunet_model import AUNET
2
3model = AUNET(input_length=30, forecast_horizon=7)
4model.fit(X_train, y_train)
5y_pred = model.predict(X_test)
