Models
===================================

Spatial Models
===================================


GAT
----------

.. autoclass:: epilearn.models.Spatial.GAT.GAT
    :members:


GCN 
----------

.. autoclass:: epilearn.models.Spatial.GCN.GCN
    :members:

GIN 
----------

.. autoclass:: epilearn.models.Spatial.GIN.GIN
    :members:

SAGE 
----------

.. autoclass:: epilearn.models.Spatial.SAGE.SAGE
    :members:

Temporal Models
===================================


ARIMA
----------

.. autoclass:: epilearn.models.Temporal.ARIMA.VARMAXModel
    :members:


DLINEAR
----------

.. autoclass:: epilearn.models.Temporal.Dlinear.DlinearModel
    :members:


GRU
----------

.. autoclass:: epilearn.models.Temporal.GRU.GRUModel
    :members:


LSTM
----------

.. autoclass:: epilearn.models.Temporal.LSTM.LSTMModel
    :members:


SIR
----------

.. autoclass:: epilearn.models.Temporal.SIR.SIR
    :members:


SIS
----------

.. autoclass:: epilearn.models.Temporal.SIR.SIS
    :members:

SEIR
----------

.. autoclass:: epilearn.models.Temporal.SIR.SEIR
    :members:

XGBOOST
----------

.. autoclass:: epilearn.models.Temporal.XGB.XGBModel
    :members:
















Spatial-Temporal Models
===================================


ATMGNN
----------

.. autoclass:: epilearn.models.SpatialTemporal.ATMGNN.ATMGNN
    :members:

MPNN_LSTM
----------

.. autoclass:: epilearn.models.SpatialTemporal.ATMGNN.MPNN_LSTM
    :members:

CNNRNN_Res
----------

.. autoclass:: epilearn.models.SpatialTemporal.CNNRNN_Res.CNNRNN_Res
    :members:


ColaGNN
----------

.. autoclass:: epilearn.models.SpatialTemporal.ColaGNN.ColaGNN
    :members:

DASTGN
----------

.. autoclass:: epilearn.models.SpatialTemporal.DASTGN.DASTGN
    :members:


DCRNN
----------

.. autoclass:: epilearn.models.SpatialTemporal.DCRNN.DCRNN
    :members:



DMP
----------

.. autoclass:: epilearn.models.SpatialTemporal.DMP.DMP
    :members:


EpiColaGNN
----------

.. autoclass:: epilearn.models.SpatialTemporal.EpiColaGNN.EpiColaGNN
    :members:


EpiGNN
----------

.. autoclass:: epilearn.models.SpatialTemporal.EpiGNN.EpiGNN
    :members:

GraphWaveNet
----------

.. autoclass:: epilearn.models.SpatialTemporal.GraphWaveNet.GraphWaveNet
    :members:




MepoGNN
----------

.. autoclass:: epilearn.models.SpatialTemporal.MepoGNN.MepoGNN
    :members:


NetSIR
----------

.. autoclass:: epilearn.models.SpatialTemporal.NetworkSIR.NetSIR
    :members:


STAN
----------

.. autoclass:: epilearn.models.SpatialTemporal.STAN.STAN
    :members:


STGCN
----------

.. autoclass:: epilearn.models.SpatialTemporal.STGCN.STGCN
    :members: