Index

A

Accuracy (class in pytorch_tabnet.metrics) append() (pytorch_tabnet.callbacks.CallbackContainer method)

assert_all_finite() (in module pytorch_tabnet.multiclass_utils) AttentiveTransformer (class in pytorch_tabnet.tab_network) AUC (class in pytorch_tabnet.metrics)

B

backward() (pytorch_tabnet.sparsemax.Entmax15Function static method) (pytorch_tabnet.sparsemax.Entmoid15 static method) (pytorch_tabnet.sparsemax.SparsemaxFunction static method)

BalancedAccuracy (class in pytorch_tabnet.metrics)

C

Callback (class in pytorch_tabnet.callbacks) CallbackContainer (class in pytorch_tabnet.callbacks) callbacks (pytorch_tabnet.callbacks.CallbackContainer attribute) cat_dims (pytorch_tabnet.abstract_model.TabModel attribute) (pytorch_tabnet.multitask.TabNetMultiTaskClassifier attribute) (pytorch_tabnet.pretraining.TabNetPretrainer attribute) (pytorch_tabnet.tab_model.TabNetClassifier attribute) (pytorch_tabnet.tab_model.TabNetRegressor attribute) cat_emb_dim (pytorch_tabnet.abstract_model.TabModel attribute) cat_idxs (pytorch_tabnet.abstract_model.TabModel attribute) (pytorch_tabnet.multitask.TabNetMultiTaskClassifier attribute) (pytorch_tabnet.pretraining.TabNetPretrainer attribute) (pytorch_tabnet.tab_model.TabNetClassifier attribute) (pytorch_tabnet.tab_model.TabNetRegressor attribute) check_classification_targets() (in module pytorch_tabnet.multiclass_utils) check_embedding_parameters() (in module pytorch_tabnet.utils) check_input() (in module pytorch_tabnet.utils)

check_list_groups() (in module pytorch_tabnet.utils) check_metrics() (in module pytorch_tabnet.metrics) check_output_dim() (in module pytorch_tabnet.multiclass_utils) check_unique_type() (in module pytorch_tabnet.multiclass_utils) check_warm_start() (in module pytorch_tabnet.utils) ClassificationSMOTE (class in pytorch_tabnet.augmentations) clip_value (pytorch_tabnet.abstract_model.TabModel attribute) ComplexEncoder (class in pytorch_tabnet.utils) compute_loss() (pytorch_tabnet.abstract_model.TabModel method) (pytorch_tabnet.multitask.TabNetMultiTaskClassifier method) (pytorch_tabnet.pretraining.TabNetPretrainer method) (pytorch_tabnet.tab_model.TabNetClassifier method) (pytorch_tabnet.tab_model.TabNetRegressor method) create_dataloaders() (in module pytorch_tabnet.pretraining_utils) (in module pytorch_tabnet.utils) create_explain_matrix() (in module pytorch_tabnet.utils) create_group_matrix() (in module pytorch_tabnet.utils) create_sampler() (in module pytorch_tabnet.utils)

D

default() (pytorch_tabnet.utils.ComplexEncoder method)

define_device() (in module pytorch_tabnet.utils) device_name (pytorch_tabnet.abstract_model.TabModel attribute)

E

early_stopping_metric (pytorch_tabnet.callbacks.EarlyStopping attribute) (pytorch_tabnet.callbacks.LRSchedulerCallback attribute) EarlyStopping (class in pytorch_tabnet.callbacks) EmbeddingGenerator (class in pytorch_tabnet.tab_network) Entmax15 (class in pytorch_tabnet.sparsemax)

entmax15() (in module pytorch_tabnet.sparsemax) Entmax15Function (class in pytorch_tabnet.sparsemax) Entmoid15 (class in pytorch_tabnet.sparsemax) entmoid15() (in module pytorch_tabnet.sparsemax) epsilon (pytorch_tabnet.abstract_model.TabModel attribute) explain() (pytorch_tabnet.abstract_model.TabModel method)

F

FeatTransformer (class in pytorch_tabnet.tab_network) filter_weights() (in module pytorch_tabnet.utils) fit() (pytorch_tabnet.abstract_model.TabModel method) (pytorch_tabnet.pretraining.TabNetPretrainer method) forward() (pytorch_tabnet.sparsemax.Entmax15 method) (pytorch_tabnet.sparsemax.Entmax15Function static method) (pytorch_tabnet.sparsemax.Entmoid15 static method) (pytorch_tabnet.sparsemax.Sparsemax method) (pytorch_tabnet.sparsemax.SparsemaxFunction static method) (pytorch_tabnet.tab_network.AttentiveTransformer method) (pytorch_tabnet.tab_network.EmbeddingGenerator method) (pytorch_tabnet.tab_network.FeatTransformer method) (pytorch_tabnet.tab_network.GBN method) (pytorch_tabnet.tab_network.GLU_Block method) (pytorch_tabnet.tab_network.GLU_Layer method) (pytorch_tabnet.tab_network.RandomObfuscator method) (pytorch_tabnet.tab_network.TabNet method) (pytorch_tabnet.tab_network.TabNetDecoder method) (pytorch_tabnet.tab_network.TabNetEncoder method) (pytorch_tabnet.tab_network.TabNetNoEmbeddings method) (pytorch_tabnet.tab_network.TabNetPretraining method)

forward_masks() (pytorch_tabnet.tab_network.TabNet method) (pytorch_tabnet.tab_network.TabNetEncoder method) (pytorch_tabnet.tab_network.TabNetNoEmbeddings method) (pytorch_tabnet.tab_network.TabNetPretraining method)

G

gamma (pytorch_tabnet.abstract_model.TabModel attribute) GBN (class in pytorch_tabnet.tab_network) get_metrics_by_names() (pytorch_tabnet.metrics.Metric class method) GLU_Block (class in pytorch_tabnet.tab_network) GLU_Layer (class in pytorch_tabnet.tab_network)

grouped_features (pytorch_tabnet.abstract_model.TabModel attribute) (pytorch_tabnet.multitask.TabNetMultiTaskClassifier attribute) (pytorch_tabnet.pretraining.TabNetPretrainer attribute) (pytorch_tabnet.tab_model.TabNetClassifier attribute) (pytorch_tabnet.tab_model.TabNetRegressor attribute)

H

History (class in pytorch_tabnet.callbacks)

I

infer_multitask_output() (in module pytorch_tabnet.multiclass_utils) infer_output_dim() (in module pytorch_tabnet.multiclass_utils) initialize_glu() (in module pytorch_tabnet.tab_network) initialize_non_glu() (in module pytorch_tabnet.tab_network)

input_dim (pytorch_tabnet.abstract_model.TabModel attribute) is_batch_level (pytorch_tabnet.callbacks.LRSchedulerCallback attribute) is_maximize (pytorch_tabnet.callbacks.EarlyStopping attribute) is_multilabel() (in module pytorch_tabnet.multiclass_utils)

L

lambda_sparse (pytorch_tabnet.abstract_model.TabModel attribute) load_class_attrs() (pytorch_tabnet.abstract_model.TabModel method) load_model() (pytorch_tabnet.abstract_model.TabModel method)

load_weights_from_unsupervised() (pytorch_tabnet.abstract_model.TabModel method) LogLoss (class in pytorch_tabnet.metrics) LRSchedulerCallback (class in pytorch_tabnet.callbacks)

M

MAE (class in pytorch_tabnet.metrics) mask_type (pytorch_tabnet.abstract_model.TabModel attribute) Metric (class in pytorch_tabnet.metrics) metric_names (pytorch_tabnet.metrics.MetricContainer attribute) (pytorch_tabnet.metrics.UnsupMetricContainer attribute)

MetricContainer (class in pytorch_tabnet.metrics) momentum (pytorch_tabnet.abstract_model.TabModel attribute) MSE (class in pytorch_tabnet.metrics)

N

n_a (pytorch_tabnet.abstract_model.TabModel attribute) n_d (pytorch_tabnet.abstract_model.TabModel attribute) n_indep_decoder (pytorch_tabnet.abstract_model.TabModel attribute)

n_independent (pytorch_tabnet.abstract_model.TabModel attribute) n_shared (pytorch_tabnet.abstract_model.TabModel attribute) n_shared_decoder (pytorch_tabnet.abstract_model.TabModel attribute) n_steps (pytorch_tabnet.abstract_model.TabModel attribute)

O

on_batch_begin() (pytorch_tabnet.callbacks.Callback method) (pytorch_tabnet.callbacks.CallbackContainer method) on_batch_end() (pytorch_tabnet.callbacks.Callback method) (pytorch_tabnet.callbacks.CallbackContainer method) (pytorch_tabnet.callbacks.History method) (pytorch_tabnet.callbacks.LRSchedulerCallback method) on_epoch_begin() (pytorch_tabnet.callbacks.Callback method) (pytorch_tabnet.callbacks.CallbackContainer method) (pytorch_tabnet.callbacks.History method) on_epoch_end() (pytorch_tabnet.callbacks.Callback method) (pytorch_tabnet.callbacks.CallbackContainer method) (pytorch_tabnet.callbacks.EarlyStopping method) (pytorch_tabnet.callbacks.History method) (pytorch_tabnet.callbacks.LRSchedulerCallback method)

on_train_begin() (pytorch_tabnet.callbacks.Callback method) (pytorch_tabnet.callbacks.CallbackContainer method) (pytorch_tabnet.callbacks.History method) on_train_end() (pytorch_tabnet.callbacks.Callback method) (pytorch_tabnet.callbacks.CallbackContainer method) (pytorch_tabnet.callbacks.EarlyStopping method) optimizer (pytorch_tabnet.callbacks.LRSchedulerCallback attribute) optimizer_fn (pytorch_tabnet.abstract_model.TabModel attribute) optimizer_params (pytorch_tabnet.abstract_model.TabModel attribute) (pytorch_tabnet.multitask.TabNetMultiTaskClassifier attribute) (pytorch_tabnet.pretraining.TabNetPretrainer attribute) (pytorch_tabnet.tab_model.TabNetClassifier attribute) (pytorch_tabnet.tab_model.TabNetRegressor attribute) output_dim (pytorch_tabnet.abstract_model.TabModel attribute)

P

patience (pytorch_tabnet.callbacks.EarlyStopping attribute) predict() (pytorch_tabnet.abstract_model.TabModel method) (pytorch_tabnet.multitask.TabNetMultiTaskClassifier method) (pytorch_tabnet.pretraining.TabNetPretrainer method) predict_func() (pytorch_tabnet.tab_model.TabNetClassifier method) (pytorch_tabnet.tab_model.TabNetRegressor method) predict_proba() (pytorch_tabnet.multitask.TabNetMultiTaskClassifier method) (pytorch_tabnet.tab_model.TabNetClassifier method) PredictDataset (class in pytorch_tabnet.utils) prefix (pytorch_tabnet.metrics.MetricContainer attribute) (pytorch_tabnet.metrics.UnsupMetricContainer attribute) prepare_target() (pytorch_tabnet.abstract_model.TabModel method) (pytorch_tabnet.multitask.TabNetMultiTaskClassifier method) (pytorch_tabnet.pretraining.TabNetPretrainer method) (pytorch_tabnet.tab_model.TabNetClassifier method) (pytorch_tabnet.tab_model.TabNetRegressor method)

pytorch_tabnet.abstract_model (module) pytorch_tabnet.augmentations (module) pytorch_tabnet.callbacks (module) pytorch_tabnet.metrics (module) pytorch_tabnet.multiclass_utils (module) pytorch_tabnet.multitask (module) pytorch_tabnet.pretraining (module) pytorch_tabnet.pretraining_utils (module) pytorch_tabnet.sparsemax (module) pytorch_tabnet.tab_model (module) pytorch_tabnet.tab_network (module) pytorch_tabnet.utils (module)

R

RandomObfuscator (class in pytorch_tabnet.tab_network) RegressionSMOTE (class in pytorch_tabnet.augmentations)

RMSE (class in pytorch_tabnet.metrics) RMSLE (class in pytorch_tabnet.metrics)

S

save_model() (pytorch_tabnet.abstract_model.TabModel method) scheduler_fn (pytorch_tabnet.abstract_model.TabModel attribute) (pytorch_tabnet.callbacks.LRSchedulerCallback attribute) scheduler_params (pytorch_tabnet.abstract_model.TabModel attribute) (pytorch_tabnet.callbacks.LRSchedulerCallback attribute) (pytorch_tabnet.multitask.TabNetMultiTaskClassifier attribute) (pytorch_tabnet.pretraining.TabNetPretrainer attribute) (pytorch_tabnet.tab_model.TabNetClassifier attribute) (pytorch_tabnet.tab_model.TabNetRegressor attribute) seed (pytorch_tabnet.abstract_model.TabModel attribute) set_params() (pytorch_tabnet.callbacks.Callback method) (pytorch_tabnet.callbacks.CallbackContainer method)

set_trainer() (pytorch_tabnet.callbacks.Callback method) (pytorch_tabnet.callbacks.CallbackContainer method) Sparsemax (class in pytorch_tabnet.sparsemax) sparsemax() (in module pytorch_tabnet.sparsemax) SparsemaxFunction (class in pytorch_tabnet.sparsemax) SparsePredictDataset (class in pytorch_tabnet.utils) SparseTorchDataset (class in pytorch_tabnet.utils) stack_batches() (pytorch_tabnet.multitask.TabNetMultiTaskClassifier method) (pytorch_tabnet.pretraining.TabNetPretrainer method) (pytorch_tabnet.tab_model.TabNetClassifier method) (pytorch_tabnet.tab_model.TabNetRegressor method)

T

TabModel (class in pytorch_tabnet.abstract_model) TabNet (class in pytorch_tabnet.tab_network) TabNetClassifier (class in pytorch_tabnet.tab_model) TabNetDecoder (class in pytorch_tabnet.tab_network) TabNetEncoder (class in pytorch_tabnet.tab_network) TabNetMultiTaskClassifier (class in pytorch_tabnet.multitask) TabNetNoEmbeddings (class in pytorch_tabnet.tab_network) TabNetPretrainer (class in pytorch_tabnet.pretraining) TabNetPretraining (class in pytorch_tabnet.tab_network) TabNetRegressor (class in pytorch_tabnet.tab_model) tol (pytorch_tabnet.callbacks.EarlyStopping attribute) TorchDataset (class in pytorch_tabnet.utils) trainer (pytorch_tabnet.callbacks.History attribute) training (pytorch_tabnet.sparsemax.Entmax15 attribute) (pytorch_tabnet.sparsemax.Sparsemax attribute) (pytorch_tabnet.tab_network.AttentiveTransformer attribute) (pytorch_tabnet.tab_network.EmbeddingGenerator attribute) (pytorch_tabnet.tab_network.FeatTransformer attribute) (pytorch_tabnet.tab_network.GBN attribute) (pytorch_tabnet.tab_network.GLU_Block attribute) (pytorch_tabnet.tab_network.GLU_Layer attribute) (pytorch_tabnet.tab_network.RandomObfuscator attribute) (pytorch_tabnet.tab_network.TabNet attribute) (pytorch_tabnet.tab_network.TabNetDecoder attribute) (pytorch_tabnet.tab_network.TabNetEncoder attribute) (pytorch_tabnet.tab_network.TabNetNoEmbeddings attribute) (pytorch_tabnet.tab_network.TabNetPretraining attribute)

type_of_target() (in module pytorch_tabnet.multiclass_utils)

U

unique_labels() (in module pytorch_tabnet.multiclass_utils) UnsupervisedLoss() (in module pytorch_tabnet.metrics) UnsupervisedLossNumpy() (in module pytorch_tabnet.metrics) UnsupervisedMetric (class in pytorch_tabnet.metrics) UnsupervisedNumpyMetric (class in pytorch_tabnet.metrics)

UnsupMetricContainer (class in pytorch_tabnet.metrics) update_fit_params() (pytorch_tabnet.abstract_model.TabModel method) (pytorch_tabnet.multitask.TabNetMultiTaskClassifier method) (pytorch_tabnet.pretraining.TabNetPretrainer method) (pytorch_tabnet.tab_model.TabNetClassifier method) (pytorch_tabnet.tab_model.TabNetRegressor method)

V

validate_eval_set() (in module pytorch_tabnet.pretraining_utils) (in module pytorch_tabnet.utils)

verbose (pytorch_tabnet.abstract_model.TabModel attribute) (pytorch_tabnet.callbacks.History attribute)

W

weight_updater() (pytorch_tabnet.tab_model.TabNetClassifier method)