Index

_

__init__() (epilearn.utils.simulation.Gravity_model method)

A

a (epilearn.models.Temporal.SIR.SEIR attribute) accuracy() (in module epilearn.utils.utils)

add_time_embedding (class in epilearn.utils.transforms) ATMGNN (class in epilearn.models.SpatialTemporal.ATMGNN)

B

beta (epilearn.models.Temporal.SIR.SEIR attribute) (epilearn.models.Temporal.SIR.SIR attribute) (epilearn.models.Temporal.SIR.SIS attribute)

C

calculate_dtw_matrix (class in epilearn.utils.transforms) CNNRNN_Res (class in epilearn.models.SpatialTemporal.CNNRNN_Res)

Compose (class in epilearn.utils.transforms) convert_to_frequency (class in epilearn.utils.transforms) cross_entropy_loss() (in module epilearn.utils.metrics)

D

decomposition (epilearn.models.Temporal.Dlinear.DlinearModel attribute) Degree_Matrix() (in module epilearn.utils.utils)

diff() (in module epilearn.utils.utils) DlinearModel (class in epilearn.models.Temporal.Dlinear)

E

edge_to_adj() (in module epilearn.utils.utils)

epi_cola_loss() (in module epilearn.utils.metrics)

F

fit() (epilearn.models.Temporal.ARIMA.VARMAXModel method) forward() (epilearn.models.Spatial.GAT.GAT method) (epilearn.models.Spatial.GCN.GCN method) (epilearn.models.Spatial.GIN.GIN method) (epilearn.models.Spatial.SAGE.SAGE method) (epilearn.models.SpatialTemporal.ATMGNN.ATMGNN method) (epilearn.models.SpatialTemporal.ATMGNN.MPNN_LSTM method) (epilearn.models.SpatialTemporal.CNNRNN_Res.CNNRNN_Res method) (epilearn.models.SpatialTemporal.STGCN.STGCN method) (epilearn.models.Temporal.Dlinear.DlinearModel method) (epilearn.models.Temporal.GRU.GRUModel method) (epilearn.models.Temporal.LSTM.LSTMModel method) (epilearn.models.Temporal.SIR.SEIR method) (epilearn.models.Temporal.SIR.SIR method) (epilearn.models.Temporal.SIR.SIS method) (epilearn.utils.transforms.add_time_embedding method) (epilearn.utils.transforms.calculate_dtw_matrix method) (epilearn.utils.transforms.convert_to_frequency method) (epilearn.utils.transforms.learnable_time_embedding method) (epilearn.utils.transforms.moving_avg method) (epilearn.utils.transforms.normalize_adj method) (epilearn.utils.transforms.normalize_feat method) (epilearn.utils.transforms.seasonality_and_trend_decompose method) (epilearn.utils.transforms.series_decomp method) (epilearn.utils.transforms.series_decomp_multi method)

from_csv() (epilearn.data.dataset.UniversalDataset class method)

G

gamma (epilearn.models.Temporal.SIR.SEIR attribute) (epilearn.models.Temporal.SIR.SIR attribute) (epilearn.models.Temporal.SIR.SIS attribute) GAT (class in epilearn.models.Spatial.GAT) GCN (class in epilearn.models.Spatial.GCN) generate_dataset() (epilearn.data.dataset.UniversalDataset method) get_ACC() (in module epilearn.utils.metrics) get_graph_from_features() (in module epilearn.utils.simulation)

get_influence() (epilearn.utils.simulation.Gravity_model method) get_loss() (in module epilearn.utils.metrics) get_MAE() (in module epilearn.utils.metrics) get_mobility_graph() (epilearn.utils.simulation.Gravity_model method) get_random_graph() (in module epilearn.utils.simulation) get_RMSE() (in module epilearn.utils.metrics) GIN (class in epilearn.models.Spatial.GIN) Gravity_model (class in epilearn.utils.simulation) GRUModel (class in epilearn.models.Temporal.GRU)

K

kronecker() (in module epilearn.utils.utils)

L

learnable_time_embedding (class in epilearn.utils.transforms) Linear_Transform (epilearn.models.Temporal.Dlinear.DlinearModel attribute)

load_from_csv() (epilearn.data.dataset.UniversalDataset class method) LSTMModel (class in epilearn.models.Temporal.LSTM)

M

moving_avg (class in epilearn.utils.transforms)

MPNN_LSTM (class in epilearn.models.SpatialTemporal.ATMGNN) mu (epilearn.models.Temporal.SIR.SEIR attribute)

N

normalize() (in module epilearn.utils.utils) normalize_adj (class in epilearn.utils.transforms)

normalize_adj() (in module epilearn.utils.utils) normalize_feat (class in epilearn.utils.transforms)

S

SAGE (class in epilearn.models.Spatial.SAGE) seasonality_and_trend_decompose (class in epilearn.utils.transforms) SEIR (class in epilearn.models.Temporal.SIR) series_decomp (class in epilearn.utils.transforms) series_decomp_multi (class in epilearn.utils.transforms)

SIR (class in epilearn.models.Temporal.SIR) SIS (class in epilearn.models.Temporal.SIR) stan_loss() (in module epilearn.utils.metrics) Static_full() (in module epilearn.utils.utils) STGCN (class in epilearn.models.SpatialTemporal.STGCN)

T

Time_geo (class in epilearn.utils.simulation)

trace_simulate() (epilearn.utils.simulation.Time_geo method)

U

UniversalDataset (class in epilearn.data.dataset)

V

VARMAXModel (class in epilearn.models.Temporal.ARIMA)