SYMBOL INDEX (115 symbols across 6 files)

FILE: transtab/dataset.py
  function load_data (line 37) | def load_data(dataname, dataset_config=None, encode_cat=False, data_cut=...
  function load_single_data (line 105) | def load_single_data(dataname, dataset_config=None, encode_cat=False, da...

FILE: transtab/evaluator.py
  function predict (line 11) | def predict(clf,
  function evaluate (line 72) | def evaluate(ypred, y_test, metric='auc', seed=123, bootstrap=False):
  function get_eval_metric_fn (line 101) | def get_eval_metric_fn(eval_metric):
  function acc_fn (line 110) | def acc_fn(y, p):
  function auc_fn (line 114) | def auc_fn(y, p):
  function mse_fn (line 117) | def mse_fn(y, p):
  class EarlyStopping (line 120) | class EarlyStopping:
    method __init__ (line 122) | def __init__(self, patience=7, verbose=False, delta=0, output_dir='ckp...
    method __call__ (line 148) | def __call__(self, val_loss, model):
    method save_checkpoint (line 170) | def save_checkpoint(self, val_loss, model):

FILE: transtab/modeling_transtab.py
  class TransTabWordEmbedding (line 21) | class TransTabWordEmbedding(nn.Module):
    method __init__ (line 25) | def __init__(self,
    method forward (line 38) | def forward(self, input_ids) -> Tensor:
  class TransTabNumEmbedding (line 44) | class TransTabNumEmbedding(nn.Module):
    method __init__ (line 48) | def __init__(self, hidden_dim) -> None:
    method forward (line 54) | def forward(self, num_col_emb, x_num_ts, num_mask=None) -> Tensor:
  class TransTabFeatureExtractor (line 64) | class TransTabFeatureExtractor:
    method __init__ (line 69) | def __init__(self,
    method __call__ (line 118) | def __call__(self, x, shuffle=False) -> Dict:
    method save (line 190) | def save(self, path):
    method load (line 211) | def load(self, path):
    method update (line 226) | def update(self, cat=None, num=None, bin=None):
    method _check_column_overlap (line 249) | def _check_column_overlap(self, cat_cols=None, num_cols=None, bin_cols...
    method _solve_duplicate_cols (line 262) | def _solve_duplicate_cols(self, duplicate_cols):
  class TransTabFeatureProcessor (line 275) | class TransTabFeatureProcessor(nn.Module):
    method __init__ (line 279) | def __init__(self,
    method _avg_embedding_by_mask (line 303) | def _avg_embedding_by_mask(self, embs, att_mask=None):
    method forward (line 311) | def forward(self,
  function _get_activation_fn (line 364) | def _get_activation_fn(activation):
  class TransTabTransformerLayer (line 375) | class TransTabTransformerLayer(nn.Module):
    method __init__ (line 377) | def __init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, ...
    method _sa_block (line 409) | def _sa_block(self, x: Tensor,
    method _ff_block (line 420) | def _ff_block(self, x: Tensor) -> Tensor:
    method __setstate__ (line 427) | def __setstate__(self, state):
    method forward (line 432) | def forward(self, src, src_mask= None, src_key_padding_mask= None, is_...
  class TransTabInputEncoder (line 458) | class TransTabInputEncoder(nn.Module):
    method __init__ (line 493) | def __init__(self,
    method forward (line 504) | def forward(self, x):
    method load (line 517) | def load(self, ckpt_dir):
  class TransTabEncoder (line 529) | class TransTabEncoder(nn.Module):
    method __init__ (line 530) | def __init__(self,
    method forward (line 567) | def forward(self, embedding, attention_mask=None, **kwargs) -> Tensor:
  class TransTabLinearClassifier (line 576) | class TransTabLinearClassifier(nn.Module):
    method __init__ (line 577) | def __init__(self,
    method forward (line 587) | def forward(self, x) -> Tensor:
  class TransTabLinearRegressor (line 593) | class TransTabLinearRegressor(nn.Module):
    method __init__ (line 594) | def __init__(self,
    method forward (line 600) | def forward(self, x) -> Tensor:
  class TransTabProjectionHead (line 606) | class TransTabProjectionHead(nn.Module):
    method __init__ (line 607) | def __init__(self,
    method forward (line 613) | def forward(self, x) -> Tensor:
  class TransTabCLSToken (line 617) | class TransTabCLSToken(nn.Module):
    method __init__ (line 620) | def __init__(self, hidden_dim) -> None:
    method expand (line 626) | def expand(self, *leading_dimensions):
    method forward (line 630) | def forward(self, embedding, attention_mask=None, **kwargs) -> Tensor:
  class TransTabModel (line 638) | class TransTabModel(nn.Module):
    method __init__ (line 683) | def __init__(self,
    method forward (line 744) | def forward(self, x, y=None):
    method load (line 771) | def load(self, ckpt_dir):
    method save (line 799) | def save(self, ckpt_dir):
    method update (line 825) | def update(self, config):
    method _check_column_overlap (line 856) | def _check_column_overlap(self, cat_cols=None, num_cols=None, bin_cols...
    method _solve_duplicate_cols (line 866) | def _solve_duplicate_cols(self, duplicate_cols):
    method _adapt_to_new_num_class (line 879) | def _adapt_to_new_num_class(self, num_class):
  class TransTabClassifier (line 891) | class TransTabClassifier(TransTabModel):
    method __init__ (line 937) | def __init__(self,
    method forward (line 974) | def forward(self, x, y=None):
  class TransTabRegressor (line 1027) | class TransTabRegressor(TransTabModel):
    method __init__ (line 1073) | def __init__(self,
    method forward (line 1108) | def forward(self, x, y=None):
  class TransTabForCL (line 1157) | class TransTabForCL(TransTabModel):
    method __init__ (line 1218) | def __init__(self,
    method forward (line 1265) | def forward(self, x, y=None):
    method _build_positive_pairs (line 1321) | def _build_positive_pairs(self, x, n):
    method cos_sim (line 1336) | def cos_sim(self, a, b):
    method self_supervised_contrastive_loss (line 1353) | def self_supervised_contrastive_loss(self, features):
    method supervised_contrastive_loss (line 1390) | def supervised_contrastive_loss(self, features, labels):

FILE: transtab/trainer.py
  class Trainer (line 24) | class Trainer:
    method __init__ (line 25) | def __init__(self,
    method train (line 101) | def train(self):
    method evaluate (line 144) | def evaluate(self):
    method train_no_dataloader (line 181) | def train_no_dataloader(self,
    method save_model (line 242) | def save_model(self, output_dir=None):
    method create_optimizer (line 264) | def create_optimizer(self):
    method create_scheduler (line 280) | def create_scheduler(self, num_training_steps, optimizer):
    method get_num_train_steps (line 289) | def get_num_train_steps(self, train_set_list, num_epoch, batch_size):
    method get_warmup_steps (line 297) | def get_warmup_steps(self, num_training_steps):
    method _build_dataloader (line 306) | def _build_dataloader(self, trainset, batch_size, collator, num_worker...

FILE: transtab/trainer_utils.py
  class TrainDataset (line 30) | class TrainDataset(Dataset):
    method __init__ (line 31) | def __init__(self, trainset):
    method __len__ (line 34) | def __len__(self):
    method __getitem__ (line 37) | def __getitem__(self, index):
  class TrainCollator (line 45) | class TrainCollator:
    method __init__ (line 48) | def __init__(self,
    method save (line 63) | def save(self, path):
    method __call__ (line 66) | def __call__(self, data):
  class SupervisedTrainCollator (line 69) | class SupervisedTrainCollator(TrainCollator):
    method __init__ (line 70) | def __init__(self,
    method __call__ (line 84) | def __call__(self, data):
  class TransTabCollatorForCL (line 90) | class TransTabCollatorForCL(TrainCollator):
    method __init__ (line 93) | def __init__(self,
    method __call__ (line 113) | def __call__(self, data):
    method _build_positive_pairs (line 132) | def _build_positive_pairs(self, x, n):
    method _build_positive_pairs_single_view (line 150) | def _build_positive_pairs_single_view(self, x):
  function get_parameter_names (line 160) | def get_parameter_names(model, forbidden_layer_types):
  function random_seed (line 175) | def random_seed(seed):
  function get_scheduler (line 181) | def get_scheduler(

FILE: transtab/transtab.py
  function build_classifier (line 14) | def build_classifier(
  function build_regressor (line 98) | def build_regressor(
  function build_extractor (line 182) | def build_extractor(
  function build_encoder (line 236) | def build_encoder(
  function build_contrastive_learner (line 329) | def build_contrastive_learner(
  function train (line 456) | def train(model,