Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/LARS-research/TabGNN
/ types & classes
Types & classes
52 in github.com/LARS-research/TabGNN
⨍
Functions
233
◇
Types & classes
52
↓ 7 callers
Class
TypeConditionalLinear
Just like torch.nn.Linear, except each input gets multiplied by different weights depending its (integer) type
models/utils.py:173
↓ 4 callers
Class
DatabaseDataset
data/DatabaseDataset.py:19
↓ 3 callers
Class
LatLongScalarEnc
data/data_encoders.py:358
↓ 2 callers
Class
ScalarPowerTransformerEnc
data/data_encoders.py:220
↓ 2 callers
Class
ScalarQuantileOrdinalEnc
data/data_encoders.py:108
↓ 2 callers
Class
ScalarQuantileTransformerEnc
data/data_encoders.py:246
↓ 2 callers
Class
ScalarRobustScalerEnc
data/data_encoders.py:196
↓ 2 callers
Class
TabularDataset
data/TabularDataset.py:14
↓ 2 callers
Class
TextSummaryScalarEnc
Returns the featuretools summary statistics about the text (num words and num_chars), but normalized
data/data_encoders.py:469
↓ 2 callers
Class
TfidfEnc
data/data_encoders.py:427
↓ 2 callers
Class
WontEncodeError
data/data_encoders.py:20
↓ 1 callers
Class
CategoricalOrdinalEnc
data/data_encoders.py:76
↓ 1 callers
Class
DatetimeScalarEnc
data/data_encoders.py:268
↓ 1 callers
Class
DummyWriter
utils.py:320
↓ 1 callers
Class
ERGATConv
models/GNN/GAT.py:179
↓ 1 callers
Class
ERGCNConv
models/GNN/GCN.py:137
↓ 1 callers
Class
EmbeddingInitializer
data/data_encoders.py:507
↓ 1 callers
Class
HANLayer
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 callers
Class
MSELoss
models/losses.py:21
↓ 1 callers
Class
NullEnc
When you want to ignore a feature
data/data_encoders.py:61
↓ 1 callers
Class
RelationalGATConv
models/GNN/GAT.py:78
↓ 1 callers
Class
SemanticAttention
models/GNN/HAN.py:12
Class
AvgPooling
models/readouts.py:9
Class
CrossEntropyLoss
models/losses.py:10
Class
DAEMLPLoss
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
Class
DatetimeOrdinalEnc
data/data_encoders.py:324
Class
ERGAT
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
Class
ERGCN
GCN using different linear mappings for each node and edge type
models/GNN/GCN.py:103
Class
EncBase
data/data_encoders.py:24
Class
FocalLoss
r""" https://arxiv.org/pdf/1708.02002.pdf
models/losses.py:28
Class
GAT
Graph Attention Network as described in https://arxiv.org/pdf/1710.10903.pdf
models/GNN/GAT.py:13
Class
GCN
Graph Convolutional Network as described in https://arxiv.org/pdf/1609.02907.pdf
models/GNN/GCN.py:12
Class
GELU
models/activations.py:10
Class
GNNModelBase
Base class for all GNN models
models/GNN/GNNModelBase.py:15
Class
GlobalAttentionPooling
models/readouts.py:29
Class
HAN
models/GNN/HAN.py:87
Class
ImbalancedDatasetSampler
Samples elements randomly from a given list of indices for imbalanced dataset Arguments: num_samples (int, optional): number of sampl
data/samplers.py:14
Class
LatLongQuantileOrdinalEnc
data/data_encoders.py:384
Class
MLMTabBERTLoss
models/losses.py:180
Class
MLMTabTransformerLoss
models/losses.py:114
Class
PoolMLP
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
Class
RelationalGAT
Relational version of Graph Attention Network
models/GNN/GAT.py:46
Class
RelationalGCN
Relational Graph Convolutional Network as described in https://arxiv.org/abs/1703.06103
models/GNN/GCN.py:68
Class
ScalarRescaleEnc
data/data_encoders.py:156
Class
Set2Set
models/readouts.py:44
Class
SetTransformerDecoder
models/readouts.py:55
Class
SortPooling
models/readouts.py:18
Class
TabLogReg
models/tabular/TabMLP.py:64
Class
TabMLP
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
Class
TabModelBase
Base class for all tabular models
models/tabular/TabModelBase.py:8
Class
TabNetSparsityLoss
r""" Loss function that augments the cross entropy loss with an encouragement for the TabNet masks to be sparse
models/losses.py:252
Class
no_GNN
models/GNN/GCN.py:55