Index

_

__best (UCTB.train.EarlyStopping.EarlyStopping attribute) (UCTB.train.EarlyStopping.EarlyStoppingTTest attribute) __p (UCTB.train.EarlyStopping.EarlyStopping attribute) __p_value_threshold (UCTB.train.EarlyStopping.EarlyStoppingTTest attribute)

__patience (UCTB.train.EarlyStopping.EarlyStopping attribute) __record_list (UCTB.train.EarlyStopping.EarlyStopping attribute) (UCTB.train.EarlyStopping.EarlyStoppingTTest attribute) __test_length (UCTB.train.EarlyStopping.EarlyStoppingTTest attribute)

A

add_ga_layer_matrix() (UCTB.model_unit.GraphModelLayers.GAL static method) add_gc_layer() (UCTB.model_unit.GraphModelLayers.GCL static method) add_multi_gc_layers() (UCTB.model_unit.GraphModelLayers.GCL static method) add_residual_ga_layer() (UCTB.model_unit.GraphModelLayers.GAL static method) adf_test() (UCTB.model.ARIMA.ARIMA static method) adjacent_to_laplacian() (UCTB.preprocess.GraphGenerator.GraphGenerator static method)

AGCRN (class in UCTB.model.AGCRN) alias_draw() (in module UCTB.model.GMAN) alias_setup() (in module UCTB.model.GMAN) AM (UCTB.preprocess.GraphGenerator.GraphGenerator attribute) ARIMA (class in UCTB.model.ARIMA) ASTGCN_submodule (class in UCTB.model.ASTGCN) attention_merge_weight() (UCTB.model_unit.GraphModelLayers.GAL static method)

B

BaseModel (class in UCTB.model_unit.BaseModel) build() (UCTB.model.DCRNN.DCRNN method) (UCTB.model.DeepST.DeepST method) (UCTB.model.GeoMAN.GeoMAN method) (UCTB.model.STMeta.STMeta method) (UCTB.model.ST_MGCN.ST_MGCN method) (UCTB.model.ST_ResNet.ST_ResNet method) (UCTB.model_unit.BaseModel.BaseModel method)

build_model() (in module UCTB.model.STGCN)

C

call() (UCTB.model_unit.DCRNN_CELL.DCGRUCell method) cheb_conv (class in UCTB.model.ASTGCN) cheb_conv_withSAt (class in UCTB.model.ASTGCN) cheb_poly_approx() (in module UCTB.model.STGCN) cheb_polynomial() (in module UCTB.model.ASTGCN)

chooseNormalizer() (in module UCTB.preprocess.preprocessor) close() (UCTB.model_unit.BaseModel.BaseModel method) compute_output_shape() (UCTB.model_unit.DCRNN_CELL.DCGRUCell method) construct_adj() (in module UCTB.model.STSGCN) correlation_adjacent() (UCTB.preprocess.GraphGenerator.GraphGenerator static method)

D

daily_slots (UCTB.dataset.data_loader.NodeTrafficLoader attribute) data (UCTB.dataset.dataset.DataSet attribute) DataSet (class in UCTB.dataset.dataset) dataset (UCTB.dataset.data_loader.NodeTrafficLoader attribute)

DCGRUCell (class in UCTB.model_unit.DCRNN_CELL) DCRNN (class in UCTB.model.DCRNN) DeepST (class in UCTB.model.DeepST) distance_adjacent() (UCTB.preprocess.GraphGenerator.GraphGenerator method)

E

EarlyStopping (class in UCTB.train.EarlyStopping)

EarlyStoppingTTest (class in UCTB.train.EarlyStopping) external_dim (UCTB.dataset.data_loader.NodeTrafficLoader attribute)

F

fit() (UCTB.model.XGBoost.XGBoost method) (UCTB.model_unit.BaseModel.BaseModel method) forward() (UCTB.model.ASTGCN.ASTGCN_submodule method) (UCTB.model.ASTGCN.Spatial_Attention_layer method) (UCTB.model.ASTGCN.cheb_conv method) (UCTB.model.ASTGCN.cheb_conv_withSAt method)

fully_con_layer() (in module UCTB.model.STGCN)

G

GAL (class in UCTB.model_unit.GraphModelLayers) gatedFusion() (in module UCTB.model.GMAN) GCL (class in UCTB.model_unit.GraphModelLayers) GCLSTMCell (class in UCTB.model_unit.ST_RNN) gcn_operation() (in module UCTB.model.STSGCN) gconv() (in module UCTB.model.STGCN) gen_batch() (in module UCTB.model.STGCN) GeoMAN (class in UCTB.model.GeoMAN)

get_adjacency_matrix() (in module UCTB.model.STSGCN) get_batch() (UCTB.train.MiniBatchTrain.MiniBatchFeedDict method) (UCTB.train.MiniBatchTrain.MiniBatchTrain method) (UCTB.train.MiniBatchTrain.MiniBatchTrainMultiData method) get_order() (UCTB.model.ARIMA.ARIMA method) GMAN() (in module UCTB.model.GMAN) GraphGenerator (class in UCTB.preprocess.GraphGenerator) gwnet (class in UCTB.model.GraphWaveNet)

H

haversine() (UCTB.preprocess.GraphGenerator.GraphGenerator static method)

HM (class in UCTB.model.HM)

I

input_transform() (in module UCTB.model.GeoMAN) interaction_adjacent() (UCTB.preprocess.GraphGenerator.GraphGenerator static method) inverse_transform() (UCTB.preprocess.preprocessor.MaxMinNormalizer method) (UCTB.preprocess.preprocessor.WhiteNormalizer method) (UCTB.preprocess.preprocessor.ZscoreNormalizer method)

is_valid_date() (in module UCTB.preprocess.time_utils) is_work_day_america() (in module UCTB.preprocess.time_utils) is_work_day_china() (in module UCTB.preprocess.time_utils)

L

layer_norm() (in module UCTB.model.STGCN) LM (UCTB.preprocess.GraphGenerator.GraphGenerator attribute)

load() (UCTB.model_unit.BaseModel.BaseModel method) load_event_scalar() (UCTB.model_unit.BaseModel.BaseModel method)

M

mae() (in module UCTB.evaluation.metric) make_concat() (UCTB.dataset.data_loader.NodeTrafficLoader method) make_model() (in module UCTB.model.ASTGCN) mape() (in module UCTB.evaluation.metric) MaxMinNormalizer (class in UCTB.preprocess.preprocessor) merge_data() (UCTB.dataset.dataset.DataSet method)

MergeIndex (UCTB.dataset.dataset.DataSet attribute) MergeWay (UCTB.dataset.dataset.DataSet attribute) MiniBatchFeedDict (class in UCTB.train.MiniBatchTrain) MiniBatchTrain (class in UCTB.train.MiniBatchTrain) MiniBatchTrainMultiData (class in UCTB.train.MiniBatchTrain) move_sample() (UCTB.preprocess.preprocessor.ST_MoveSample method) multiple_process() (in module UCTB.utils.multi_threads)

N

node_monthly_interaction (UCTB.dataset.dataset.DataSet attribute) node_station_info (UCTB.dataset.dataset.DataSet attribute)

node_traffic (UCTB.dataset.dataset.DataSet attribute) NodeTrafficLoader (class in UCTB.dataset.data_loader) Normalizer (class in UCTB.preprocess.preprocessor)

O

output_layer() (in module UCTB.model.STGCN) (in module UCTB.model.STSGCN)

output_size (UCTB.model_unit.DCRNN_CELL.DCGRUCell attribute)

P

position_embedding() (in module UCTB.model.STSGCN) predict() (UCTB.model.ARIMA.ARIMA method) (UCTB.model.HM.HM method) (UCTB.model.XGBoost.XGBoost method) (UCTB.model_unit.BaseModel.BaseModel method)

R

restart() (UCTB.train.MiniBatchTrain.MiniBatchFeedDict method) (UCTB.train.MiniBatchTrain.MiniBatchTrain method) (UCTB.train.MiniBatchTrain.MiniBatchTrainMultiData method)

rmse() (in module UCTB.evaluation.metric)

S

save() (UCTB.model_unit.BaseModel.BaseModel method) scaled_Laplacian_ASTGCN() (in module UCTB.preprocess.GraphGenerator) scaled_laplacian_STGCN() (in module UCTB.preprocess.GraphGenerator) shuffle() (UCTB.train.MiniBatchTrain.MiniBatchFeedDict static method) (UCTB.train.MiniBatchTrain.MiniBatchTrain static method) (UCTB.train.MiniBatchTrain.MiniBatchTrainMultiData static method) smape() (in module UCTB.evaluation.metric) Spatial_Attention_layer (class in UCTB.model.ASTGCN) spatialAttention() (in module UCTB.model.GMAN) spatio_conv_layer() (in module UCTB.model.STGCN) split_data() (UCTB.preprocess.preprocessor.SplitData static method) split_feed_dict() (UCTB.preprocess.preprocessor.SplitData static method) split_timesteps() (in module UCTB.model.GeoMAN) SplitData (class in UCTB.preprocess.preprocessor)

st_conv_block() (in module UCTB.model.STGCN) ST_MGCN (class in UCTB.model.ST_MGCN) ST_MoveSample (class in UCTB.preprocess.preprocessor) ST_ResNet (class in UCTB.model.ST_ResNet) state_size (UCTB.model_unit.DCRNN_CELL.DCGRUCell attribute) station_number (UCTB.dataset.data_loader.NodeTrafficLoader attribute) STEmbedding() (in module UCTB.model.GMAN) sthgcn_layer_individual() (in module UCTB.model.STSGCN) sthgcn_layer_sharing() (in module UCTB.model.STSGCN) STMeta (class in UCTB.model.STMeta) stop() (UCTB.train.EarlyStopping.EarlyStopping method) (UCTB.train.EarlyStopping.EarlyStoppingTTest method) stsgcl() (in module UCTB.model.STSGCN) stsgcm() (in module UCTB.model.STSGCN) stsgcn() (in module UCTB.model.STSGCN)

T

temporal_conv_layer() (in module UCTB.model.STGCN) temporalAttention() (in module UCTB.model.GMAN) time_fitness (UCTB.dataset.dataset.DataSet attribute) time_range (UCTB.dataset.dataset.DataSet attribute) train_closeness (UCTB.dataset.data_loader.NodeTrafficLoader attribute) train_y (UCTB.dataset.data_loader.NodeTrafficLoader attribute)

transform() (UCTB.preprocess.preprocessor.MaxMinNormalizer method) (UCTB.preprocess.preprocessor.WhiteNormalizer method) (UCTB.preprocess.preprocessor.ZscoreNormalizer method) transformAttention() (in module UCTB.model.GMAN) trunc_mae() (in module UCTB.evaluation.metric) trunc_rmse() (in module UCTB.evaluation.metric) trunc_smape() (in module UCTB.evaluation.metric)

U

UCTB (module) UCTB.dataset.data_loader (module) UCTB.dataset.dataset (module) UCTB.evaluation.metric (module) UCTB.model.AGCRN (module) UCTB.model.ARIMA (module) UCTB.model.ASTGCN (module) UCTB.model.DCRNN (module) UCTB.model.DeepST (module) UCTB.model.GeoMAN (module) UCTB.model.GMAN (module) UCTB.model.GraphWaveNet (module) UCTB.model.HM (module) UCTB.model.ST_MGCN (module)

UCTB.model.ST_ResNet (module) UCTB.model.STGCN (module) UCTB.model.STMeta (module) UCTB.model.STSGCN (module) UCTB.model.XGBoost (module) UCTB.model_unit.BaseModel (module) UCTB.model_unit.DCRNN_CELL (module) UCTB.model_unit.GraphModelLayers (module) UCTB.model_unit.ST_RNN (module) UCTB.preprocess.GraphGenerator (module) UCTB.preprocess.preprocessor (module) UCTB.preprocess.time_utils (module) UCTB.train.EarlyStopping (module) UCTB.train.MiniBatchTrain (module) UCTB.utils.multi_threads (module)

V

variable_summaries() (in module UCTB.model.STGCN)

W

WhiteNormalizer (class in UCTB.preprocess.preprocessor)

X

XGBoost (class in UCTB.model.XGBoost)

Z

ZscoreNormalizer (class in UCTB.preprocess.preprocessor)