spateo.external.CAST.model.aug

Functions

__random_aug(graph, x, feat_drop_rate, edge_mask_rate)

__drop_feature(x, drop_prob)

__mask_edge(graph, mask_prob)

random_aug(graph, x, feat_drop_rate, edge_mask_rate)

drop_feature(x, drop_prob)

mask_edge(graph, mask_prob)

Module Contents

spateo.external.CAST.model.aug.__random_aug(graph, x, feat_drop_rate, edge_mask_rate)[source]
spateo.external.CAST.model.aug.__drop_feature(x, drop_prob)[source]
spateo.external.CAST.model.aug.__mask_edge(graph, mask_prob)[source]
spateo.external.CAST.model.aug.random_aug(graph, x, feat_drop_rate, edge_mask_rate)[source]
spateo.external.CAST.model.aug.drop_feature(x, drop_prob)[source]
spateo.external.CAST.model.aug.mask_edge(graph, mask_prob)[source]