Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/NonameUntitled/tiger
/ functions
Functions
55 in github.com/NonameUntitled/tiger
⨍
Functions
55
◇
Types & classes
18
↓ 6 callers
Method
load
(self, checkpoint)
tiger/modeling/trainer/trainer.py:146
↓ 5 callers
Function
create_masked_tensor
(data, lengths, is_right_aligned=False)
tiger/modeling/utils/__init__.py:77
↓ 4 callers
Function
create_logger
( name, level=logging.DEBUG, format='[%(asctime)s] [%(levelname)s]: %(message)s',
tiger/modeling/utils/__init__.py:46
↓ 3 callers
Method
create
(cls, inter_json_path, max_sequence_length, sampler_type, is_extended=False)
tiger/modeling/dataset/base.py:25
↓ 3 callers
Method
eval
(self)
tiger/modeling/trainer/trainer.py:135
↓ 2 callers
Method
_init_weights
(self, initializer_range)
tiger/modeling/models/base.py:7
↓ 2 callers
Function
fix_random_seed
(seed)
tiger/modeling/utils/__init__.py:57
↓ 2 callers
Method
get_samplers
(self)
tiger/modeling/dataset/base.py:90
↓ 2 callers
Function
parse_args
()
tiger/modeling/utils/__init__.py:35
↓ 2 callers
Method
save
(self)
tiger/modeling/trainer/trainer.py:140
↓ 2 callers
Method
train
(self)
tiger/modeling/trainer/trainer.py:72
↓ 1 callers
Method
__init__
( self, positive_prefix, negative_prefix, output_prefix=None
tiger/modeling/loss/base.py:6
↓ 1 callers
Method
_apply_sequential_encoder
(self, events, lengths)
tiger/modeling/models/sasrec.py:122
↓ 1 callers
Method
add_scalar
(self, *args, **kwargs)
tiger/modeling/utils/__init__.py:31
↓ 1 callers
Function
main
()
tiger/train_tiger.py:20
↓ 1 callers
Function
main
()
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
Method
create
(cls, mapping_path, sem_id_len, user_ids_count)
tiger/modeling/dataloader/batch_processors.py:18
Method
dataset
(self)
tiger/modeling/dataset/samplers.py:53
Method
forward
(self, inputs)
tiger/modeling/models/tiger.py:132
Method
forward
(self, inputs)
tiger/modeling/models/sasrec.py:52
Method
forward
(self, inputs)
tiger/modeling/loss/base.py:17
Method
forward
(self, inputs)
tiger/modeling/loss/base.py:40
Function
get_activation_function
(name, **kwargs)
tiger/modeling/utils/__init__.py:64
Method
max_sequence_length
(self)
tiger/modeling/dataset/base.py:98
Method
num_items
(self)
tiger/modeling/dataset/base.py:94