MCPcopy Create free account

hub / github.com/LARS-research/TabGNN / types & classes

Types & classes52 in github.com/LARS-research/TabGNN

↓ 7 callersClassTypeConditionalLinear
Just like torch.nn.Linear, except each input gets multiplied by different weights depending its (integer) type
models/utils.py:173
↓ 4 callersClassDatabaseDataset
data/DatabaseDataset.py:19
↓ 3 callersClassLatLongScalarEnc
data/data_encoders.py:358
↓ 2 callersClassScalarPowerTransformerEnc
data/data_encoders.py:220
↓ 2 callersClassScalarQuantileOrdinalEnc
data/data_encoders.py:108
↓ 2 callersClassScalarQuantileTransformerEnc
data/data_encoders.py:246
↓ 2 callersClassScalarRobustScalerEnc
data/data_encoders.py:196
↓ 2 callersClassTabularDataset
data/TabularDataset.py:14
↓ 2 callersClassTextSummaryScalarEnc
Returns the featuretools summary statistics about the text (num words and num_chars), but normalized
data/data_encoders.py:469
↓ 2 callersClassTfidfEnc
data/data_encoders.py:427
↓ 2 callersClassWontEncodeError
data/data_encoders.py:20
↓ 1 callersClassCategoricalOrdinalEnc
data/data_encoders.py:76
↓ 1 callersClassDatetimeScalarEnc
data/data_encoders.py:268
↓ 1 callersClassDummyWriter
utils.py:320
↓ 1 callersClassERGATConv
models/GNN/GAT.py:179
↓ 1 callersClassERGCNConv
models/GNN/GCN.py:137
↓ 1 callersClassEmbeddingInitializer
data/data_encoders.py:507
↓ 1 callersClassHANLayer
HAN layer. Arguments --------- meta_paths : list of metapaths, each as a list of edge types in_size : input feature dimension
models/GNN/HAN.py:29
↓ 1 callersClassMSELoss
models/losses.py:21
↓ 1 callersClassNullEnc
When you want to ignore a feature
data/data_encoders.py:61
↓ 1 callersClassRelationalGATConv
models/GNN/GAT.py:78
↓ 1 callersClassSemanticAttention
models/GNN/HAN.py:12
ClassAvgPooling
models/readouts.py:9
ClassCrossEntropyLoss
models/losses.py:10
ClassDAEMLPLoss
Denoising AutoEncoder (DAE) Loss. Tries to reconstruct the input with an MLP, and penalizes the difference. Intended for use in pretraining
models/losses.py:51
ClassDatetimeOrdinalEnc
data/data_encoders.py:324
ClassERGAT
GAT using different linear mappings for each node and edge type todo: compare to this relational GAT model: https://openreview.net/pdf?id=Bk
models/GNN/GAT.py:145
ClassERGCN
GCN using different linear mappings for each node and edge type
models/GNN/GCN.py:103
ClassEncBase
data/data_encoders.py:24
ClassFocalLoss
r""" https://arxiv.org/pdf/1708.02002.pdf
models/losses.py:28
ClassGAT
Graph Attention Network as described in https://arxiv.org/pdf/1710.10903.pdf
models/GNN/GAT.py:13
ClassGCN
Graph Convolutional Network as described in https://arxiv.org/pdf/1609.02907.pdf
models/GNN/GCN.py:12
ClassGELU
models/activations.py:10
ClassGNNModelBase
Base class for all GNN models
models/GNN/GNNModelBase.py:15
ClassGlobalAttentionPooling
models/readouts.py:29
ClassHAN
models/GNN/HAN.py:87
ClassImbalancedDatasetSampler
Samples elements randomly from a given list of indices for imbalanced dataset Arguments: num_samples (int, optional): number of sampl
data/samplers.py:14
ClassLatLongQuantileOrdinalEnc
data/data_encoders.py:384
ClassMLMTabBERTLoss
models/losses.py:180
ClassMLMTabTransformerLoss
models/losses.py:114
ClassPoolMLP
Model that ignores relational structure and just inits the nodes, pools them, and computes the output. Inspired by https://arxiv.org/pdf/1905
models/GNN/PoolMLP.py:6
ClassRelationalGAT
Relational version of Graph Attention Network
models/GNN/GAT.py:46
ClassRelationalGCN
Relational Graph Convolutional Network as described in https://arxiv.org/abs/1703.06103
models/GNN/GCN.py:68
ClassScalarRescaleEnc
data/data_encoders.py:156
ClassSet2Set
models/readouts.py:44
ClassSetTransformerDecoder
models/readouts.py:55
ClassSortPooling
models/readouts.py:18
ClassTabLogReg
models/tabular/TabMLP.py:64
ClassTabMLP
Straightforward MLP model for tabular data, loosely based on github.com/fastai/fastai/blob/master/fastai/tabular layer_sizes can contain int
models/tabular/TabMLP.py:7
ClassTabModelBase
Base class for all tabular models
models/tabular/TabModelBase.py:8
ClassTabNetSparsityLoss
r""" Loss function that augments the cross entropy loss with an encouragement for the TabNet masks to be sparse
models/losses.py:252
Classno_GNN
models/GNN/GCN.py:55