EpiLearn
Installation
Installation
Usage
Quickstart for EpiLearn
Pipeline for Epidemic Modling
Simulation
Utilities
Package API
Dataset
Models
Spatial Models
Temporal Models
Spatial-Temporal Models
Tasks
Utils
Utility_Functions
Metrics
Simulation
Transformation
Utils
Visualization
EpiLearn
Index
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
K
|
L
|
M
|
N
|
P
|
S
|
T
|
U
|
V
|
X
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)
ColaGNN (class in epilearn.models.SpatialTemporal.ColaGNN)
Compose (class in epilearn.utils.transforms)
convert_to_frequency (class in epilearn.utils.transforms)
cross_entropy_loss() (in module epilearn.utils.metrics)
D
DASTGN (class in epilearn.models.SpatialTemporal.DASTGN)
DCRNN (class in epilearn.models.SpatialTemporal.DCRNN)
decoder() (epilearn.models.SpatialTemporal.DCRNN.DCRNN method)
decomposition (epilearn.models.Temporal.Dlinear.DlinearModel attribute)
Degree_Matrix() (in module epilearn.utils.utils)
Detection (class in epilearn.tasks.detection)
diff() (in module epilearn.utils.utils)
DlinearModel (class in epilearn.models.Temporal.Dlinear)
DMP (class in epilearn.models.SpatialTemporal.DMP)
download() (epilearn.data.dataset.UniversalDataset method)
E
edge_to_adj() (in module epilearn.utils.utils)
encoder() (epilearn.models.SpatialTemporal.DCRNN.DCRNN method)
epi_cola_loss() (in module epilearn.utils.metrics)
EpiColaGNN (class in epilearn.models.SpatialTemporal.EpiColaGNN)
EpiGNN (class in epilearn.models.SpatialTemporal.EpiGNN)
evaluate_model() (epilearn.tasks.detection.Detection method)
(epilearn.tasks.forecast.Forecast method)
F
fit() (epilearn.models.Temporal.ARIMA.VARMAXModel method)
(epilearn.models.Temporal.XGB.XGBModel method)
Forecast (class in epilearn.tasks.forecast)
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.ColaGNN.ColaGNN method)
(epilearn.models.SpatialTemporal.DASTGN.DASTGN method)
(epilearn.models.SpatialTemporal.DCRNN.DCRNN method)
(epilearn.models.SpatialTemporal.DMP.DMP method)
(epilearn.models.SpatialTemporal.EpiColaGNN.EpiColaGNN method)
(epilearn.models.SpatialTemporal.EpiGNN.EpiGNN method)
(epilearn.models.SpatialTemporal.GraphWaveNet.GraphWaveNet method)
(epilearn.models.SpatialTemporal.MepoGNN.MepoGNN method)
(epilearn.models.SpatialTemporal.NetworkSIR.NetSIR method)
(epilearn.models.SpatialTemporal.STAN.STAN 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)
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_loss() (in module epilearn.utils.metrics)
get_MAE() (in module epilearn.utils.metrics)
get_RMSE() (in module epilearn.utils.metrics)
get_splits() (epilearn.tasks.detection.Detection method)
(epilearn.tasks.forecast.Forecast method)
GIN (class in epilearn.models.Spatial.GIN)
GraphWaveNet (class in epilearn.models.SpatialTemporal.GraphWaveNet)
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)
LSTMModel (class in epilearn.models.Temporal.LSTM)
M
MepoGNN (class in epilearn.models.SpatialTemporal.MepoGNN)
moving_avg (class in epilearn.utils.transforms)
MPNN_LSTM (class in epilearn.models.SpatialTemporal.ATMGNN)
mu (epilearn.models.Temporal.SIR.SEIR attribute)
N
NetSIR (class in epilearn.models.SpatialTemporal.NetworkSIR)
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)
P
plot_graph() (in module epilearn.visualize.plot_series)
plot_series() (in module epilearn.visualize.plot_series)
S
SAGE (class in epilearn.models.Spatial.SAGE)
save() (epilearn.data.dataset.UniversalDataset method)
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 (class in epilearn.models.SpatialTemporal.STAN)
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)
train_model() (epilearn.tasks.detection.Detection method)
(epilearn.tasks.forecast.Forecast method)
U
UniversalDataset (class in epilearn.data.dataset)
V
VARMAXModel (class in epilearn.models.Temporal.ARIMA)
X
XGBModel (class in epilearn.models.Temporal.XGB)