MCPcopy Create free account

hub / github.com/NonameUntitled/tiger / functions

Functions55 in github.com/NonameUntitled/tiger

↓ 6 callersMethodload
(self, checkpoint)
tiger/modeling/trainer/trainer.py:146
↓ 5 callersFunctioncreate_masked_tensor
(data, lengths, is_right_aligned=False)
tiger/modeling/utils/__init__.py:77
↓ 4 callersFunctioncreate_logger
( name, level=logging.DEBUG, format='[%(asctime)s] [%(levelname)s]: %(message)s',
tiger/modeling/utils/__init__.py:46
↓ 3 callersMethodcreate
(cls, inter_json_path, max_sequence_length, sampler_type, is_extended=False)
tiger/modeling/dataset/base.py:25
↓ 3 callersMethodeval
(self)
tiger/modeling/trainer/trainer.py:135
↓ 2 callersMethod_init_weights
(self, initializer_range)
tiger/modeling/models/base.py:7
↓ 2 callersFunctionfix_random_seed
(seed)
tiger/modeling/utils/__init__.py:57
↓ 2 callersMethodget_samplers
(self)
tiger/modeling/dataset/base.py:90
↓ 2 callersFunctionparse_args
()
tiger/modeling/utils/__init__.py:35
↓ 2 callersMethodsave
(self)
tiger/modeling/trainer/trainer.py:140
↓ 2 callersMethodtrain
(self)
tiger/modeling/trainer/trainer.py:72
↓ 1 callersMethod__init__
( self, positive_prefix, negative_prefix, output_prefix=None
tiger/modeling/loss/base.py:6
↓ 1 callersMethod_apply_sequential_encoder
(self, events, lengths)
tiger/modeling/models/sasrec.py:122
↓ 1 callersMethodadd_scalar
(self, *args, **kwargs)
tiger/modeling/utils/__init__.py:31
↓ 1 callersFunctionmain
()
tiger/train_tiger.py:20
↓ 1 callersFunctionmain
()
tiger/train_sasrec.py:19
Method__call__
(self, batch)
tiger/modeling/dataloader/batch_processors.py:29
Method__call__
(self, inputs, pred_prefix, labels_prefix)
tiger/modeling/metric/base.py:8
Method__call__
(self, inputs, pred_prefix, labels_prefix)
tiger/modeling/metric/base.py:27
Method__call__
(self, inputs, pred_prefix, labels_prefix)
tiger/modeling/metric/base.py:48
Method__call__
(self, inputs, pred_prefix, labels_prefix)
tiger/modeling/metric/base.py:66
Method__call__
(self, key, value, step_num, prefix)
tiger/modeling/trainer/callbacks.py:14
Method__call__
(self, step_num)
tiger/modeling/trainer/callbacks.py:43
Method__call__
(self, input_ids: torch.LongTensor, scores: torch.FloatTensor)
tiger/modeling/models/tiger.py:40
Method__getitem__
(self, index)
tiger/modeling/dataset/samplers.py:39
Method__getitem__
(self, index)
tiger/modeling/dataset/samplers.py:59
Method__init__
( self, experiment_name, logs_dir='../tensorboard_logs', use_t
tiger/modeling/utils/__init__.py:16
Method__init__
(self, dataset, prediction_type, max_sequence_length)
tiger/modeling/dataset/samplers.py:5
Method__init__
(self, dataset, max_sequence_length)
tiger/modeling/dataset/samplers.py:48
Method__init__
( self, train_sampler, validation_sampler, test_sampler,
tiger/modeling/dataset/base.py:10
Method__init__
(self, semantic_ids=None, sem_id_len=None, user_ids_count=None)
tiger/modeling/dataloader/batch_processors.py:8
Method__init__
(self, k)
tiger/modeling/metric/base.py:5
Method__init__
(self, k, codebook_size, num_codebooks)
tiger/modeling/metric/base.py:22
Method__init__
(self, k)
tiger/modeling/metric/base.py:45
Method__init__
(self, k, codebook_size, num_codebooks)
tiger/modeling/metric/base.py:61
Method__init__
(self, tensorboard_writer, on_step)
tiger/modeling/trainer/callbacks.py:10
Method__init__
( self, tensorboard_writer, step_name, model, data
tiger/modeling/trainer/callbacks.py:23
Method__init__
( self, experiment_name, train_dataloader, validation_dataload
tiger/modeling/trainer/trainer.py:13
Method__init__
(self, num_codebooks, codebook_size, index_path, num_beams, visited_items)
tiger/modeling/models/tiger.py:11
Method__init__
( self, embedding_dim, codebook_size, sem_id_len,
tiger/modeling/models/tiger.py:72
Method__init__
( self, num_items, max_sequence_length, embedding_dim,
tiger/modeling/models/sasrec.py:9
Method__init__
(self, predictions_prefix, output_prefix=None)
tiger/modeling/loss/base.py:35
Method__len__
(self)
tiger/modeling/dataset/samplers.py:43
Method__len__
(self)
tiger/modeling/dataset/samplers.py:56
Method_all_items_transform
(self, sample)
tiger/modeling/dataset/samplers.py:15
Method_last_item_transform
(self, sample)
tiger/modeling/dataset/samplers.py:27
Methodcreate
(cls, mapping_path, sem_id_len, user_ids_count)
tiger/modeling/dataloader/batch_processors.py:18
Methoddataset
(self)
tiger/modeling/dataset/samplers.py:53
Methodforward
(self, inputs)
tiger/modeling/models/tiger.py:132
Methodforward
(self, inputs)
tiger/modeling/models/sasrec.py:52
Methodforward
(self, inputs)
tiger/modeling/loss/base.py:17
Methodforward
(self, inputs)
tiger/modeling/loss/base.py:40
Functionget_activation_function
(name, **kwargs)
tiger/modeling/utils/__init__.py:64
Methodmax_sequence_length
(self)
tiger/modeling/dataset/base.py:98
Methodnum_items
(self)
tiger/modeling/dataset/base.py:94