LICENSE
README.md
setup.py
epilearn/__init__.py
epilearn.egg-info/PKG-INFO
epilearn.egg-info/SOURCES.txt
epilearn.egg-info/dependency_links.txt
epilearn.egg-info/requires.txt
epilearn.egg-info/top_level.txt
epilearn/data/__init__.py
epilearn/data/base.py
epilearn/data/dataset.py
epilearn/models/__init__.py
epilearn/models/General/__init__.py
epilearn/models/General/mlp.py
epilearn/models/Spatial/GAT.py
epilearn/models/Spatial/GCN.py
epilearn/models/Spatial/GIN.py
epilearn/models/Spatial/SAGE.py
epilearn/models/Spatial/__init__.py
epilearn/models/Spatial/base.py
epilearn/models/SpatialTemporal/ATMGNN.py
epilearn/models/SpatialTemporal/CNNRNN_Res.py
epilearn/models/SpatialTemporal/ColaGNN.py
epilearn/models/SpatialTemporal/DASTGN.py
epilearn/models/SpatialTemporal/DCRNN.py
epilearn/models/SpatialTemporal/DMP.py
epilearn/models/SpatialTemporal/EpiColaGNN.py
epilearn/models/SpatialTemporal/EpiGNN.py
epilearn/models/SpatialTemporal/GraphLSTM.py
epilearn/models/SpatialTemporal/GraphWaveNet.py
epilearn/models/SpatialTemporal/HierST.py
epilearn/models/SpatialTemporal/MepoGNN.py
epilearn/models/SpatialTemporal/NetworkSIR.py
epilearn/models/SpatialTemporal/STAN.py
epilearn/models/SpatialTemporal/STGCN.py
epilearn/models/SpatialTemporal/__init__.py
epilearn/models/SpatialTemporal/base.py
epilearn/models/Temporal/ARIMA.py
epilearn/models/Temporal/Dlinear.py
epilearn/models/Temporal/EINN.py
epilearn/models/Temporal/GRU.py
epilearn/models/Temporal/LSTM.py
epilearn/models/Temporal/SIR.py
epilearn/models/Temporal/XGB.py
epilearn/models/Temporal/__init__.py
epilearn/models/Temporal/base.py
epilearn/tasks/__init__.py
epilearn/tasks/base.py
epilearn/tasks/detection.py
epilearn/tasks/forecast.py
epilearn/tasks/projection.py
epilearn/tasks/surveillance.py
epilearn/utils/__init__.py
epilearn/utils/metrics.py
epilearn/utils/simulation.py
epilearn/utils/transforms.py
epilearn/utils/utils.py
epilearn/visualize/__init__.py
epilearn/visualize/plot.py
tests/test.py
tests/test_cola.py
tests/test_dastgn.py
tests/test_epi.py
tests/test_epi_cola.py
tests/test_hierst.py
tests/test_mepo.py
tests/test_spatial.py
tests/test_stan.py