MCPcopy Create free account

hub / github.com/HonghuiBao2000/LETTER / functions

Functions205 in github.com/HonghuiBao2000/LETTER

⨍ Functions 205◇ Types & classes 22
↓ 1 callersFunctionparse_test_args
(parser)
LETTER-TIGER/utils.py:79
↓ 1 callersFunctionparse_train_args
(parser)
LETTER-LC-Rec/utils.py:59
↓ 1 callersFunctionparse_train_args
(parser)
LETTER-TIGER/utils.py:54
↓ 1 callersFunctionprefix_allowed_tokens_fn
(candidate_trie)
LETTER-TIGER/generation_trie.py:91
↓ 1 callersFunctionprefix_allowed_tokens_fn
(candidate_trie)
LETTER-TIGER/utils.py:161
↓ 1 callersFunctionpreprocess_text
(args)
data_process/amazon_text_emb.py:38
↓ 1 callersMethodranking_loss
(self, shift_logits, shift_labels)
LETTER-LC-Rec/modeling_letter.py:33
↓ 1 callersMethodranking_loss
(self, lm_logits, labels)
LETTER-TIGER/modeling_letter.py:37
↓ 1 callersFunctionset_device
(gpu_id)
data_process/utils.py:75
↓ 1 callersMethodset_hyper
(self, temperature)
LETTER-LC-Rec/modeling_letter.py:30
↓ 1 callersMethodset_hyper
(self,temperature)
LETTER-TIGER/modeling_letter.py:33
↓ 1 callersMethodset_prompt
(self, prompt_id)
LETTER-LC-Rec/data.py:206
↓ 1 callersMethodset_prompt
(self, prompt_id)
LETTER-TIGER/data.py:239
↓ 1 callersFunctiontest
(args)
LETTER-TIGER/test.py:21
↓ 1 callersFunctiontest_ddp
(args)
LETTER-LC-Rec/test_ddp.py:26
↓ 1 callersMethodtotal_loss
(self, shift_logits, shift_labels)
LETTER-LC-Rec/modeling_letter.py:42
↓ 1 callersMethodtotal_loss
(self, lm_logits, labels, decoder_input_ids)
LETTER-TIGER/modeling_letter.py:47
↓ 1 callersFunctiontrain
(args)
LETTER-LC-Rec/lora_finetune.py:30
↓ 1 callersFunctiontrain
(args)
LETTER-TIGER/finetune.py:18
↓ 1 callersMethodvq_ini
(self, x)
RQ-VAE/models/rq.py:34
↓ 1 callersMethodvq_init
(self)
RQ-VAE/trainer.py:101
↓ 1 callersMethodvq_init
(self, x, use_sk=True)
RQ-VAE/models/vq.py:122
↓ 1 callersMethodvq_initialization
(self,x, use_sk=True)
RQ-VAE/models/rqvae.py:82
Method__call__
(self, batch)
LETTER-LC-Rec/collator.py:23
Method__call__
(self, batch)
LETTER-LC-Rec/collator.py:63
Method__call__
(self, batch)
LETTER-TIGER/collator.py:24
Method__call__
(self, batch)
LETTER-TIGER/collator.py:57
Method__getitem__
(self, index)
LETTER-LC-Rec/data.py:252
Method__getitem__
(self, index)
LETTER-LC-Rec/data.py:457
Method__getitem__
(self, index)
LETTER-LC-Rec/data.py:535
Method__getitem__
(self, index)
LETTER-LC-Rec/data.py:645
Method__getitem__
(self, index)
LETTER-LC-Rec/data.py:747
Method__getitem__
(self, index)
LETTER-LC-Rec/data.py:839
Method__getitem__
(self, index)
RQ-VAE/datasets.py:15
Method__getitem__
(self, value)
LETTER-TIGER/generation_trie.py:87
Method__getitem__
(self, index)
LETTER-TIGER/data.py:247
Method__init__
(self, args, tokenizer)
LETTER-LC-Rec/collator.py:15
Method__init__
(self, args, tokenizer)
LETTER-LC-Rec/collator.py:53
Method__init__
(self, args, mode="train", prompt_sample_num=1, prompt_id=0, sample_num=-1)
LETTER-LC-Rec/data.py:99
Method__init__
(self, args, mode="train", prompt_sample_num=1, prompt_id=0, sample_num=-1)
LETTER-LC-Rec/data.py:277
Method__init__
(self, args, task="item2index", prompt_sample_num=1, sample_num=-1)
LETTER-LC-Rec/data.py:480
Method__init__
(self, args, mode="train", prompt_sample_num=1, prompt_id=0, sample_num=-1)
LETTER-LC-Rec/data.py:551
Method__init__
(self, args, prompt_sample_num=1, sample_num=-1)
LETTER-LC-Rec/data.py:669
Method__init__
(self, args, prompt_id=0, sample_num=-1)
LETTER-LC-Rec/data.py:768
Method__init__
(self, config)
LETTER-LC-Rec/modeling_letter.py:26
Method__init__
(self,data_path)
RQ-VAE/datasets.py:8
Method__init__
(self, args, model)
RQ-VAE/trainer.py:19
Method__init__
(self, n_e, e_dim, mu = 0.25, beta = 1, kmeans_init = False, kmeans_iters = 10,
RQ-VAE/models/vq.py:11
Method__init__
(self, n_e_list, e_dim, sk_epsilons, beta = 1, kmeans_init = False, kmeans_iters = 100, sk_it
RQ-VAE/models/rq.py:9
Method__init__
( self, layers, dropout=0.0, activation="relu", bn=False )
RQ-VAE/models/layers.py:9
Method__init__
(self, in_dim=768, num_emb_list=None, e_dim=64,
RQ-VAE/models/rqvae.py:14
Method__init__
(self, args, tokenizer)
LETTER-TIGER/collator.py:16
Method__init__
(self, args, tokenizer)
LETTER-TIGER/collator.py:51
Method__init__
(self, sequences: List[List[int]] = [])
LETTER-TIGER/generation_trie.py:8
Method__init__
(self, args, mode="train", prompt_sample_num=1, prompt_id=0, sample_num=-1)
LETTER-TIGER/data.py:109
Method__init__
(self, config: T5Config)
LETTER-TIGER/modeling_letter.py:26
Method__iter__
(self)
LETTER-TIGER/generation_trie.py:72
Method__len__
(self)
LETTER-LC-Rec/data.py:210
Method__len__
(self)
LETTER-LC-Rec/data.py:415
Method__len__
(self)
LETTER-LC-Rec/data.py:522
Method__len__
(self)
LETTER-LC-Rec/data.py:623
Method__len__
(self)
LETTER-LC-Rec/data.py:733
Method__len__
(self)
LETTER-LC-Rec/data.py:826
Method__len__
(self)
RQ-VAE/datasets.py:20
Method__len__
(self)
LETTER-TIGER/generation_trie.py:84
Method__len__
(self)
LETTER-TIGER/data.py:243
Method_load_data
(self)
LETTER-LC-Rec/data.py:38
Method_load_data
(self)
LETTER-TIGER/data.py:38
Method_process_data
(self)
LETTER-LC-Rec/data.py:91
Method_process_data
(self)
LETTER-TIGER/data.py:101
Method_traverse
(prefix_sequence, trie_dict)
LETTER-TIGER/generation_trie.py:73
Functioncheck_path
(path)
data_process/utils.py:70
Methodforward
r""" Args: labels (`torch.LongTensor` of shape `(batch_size, sequence_length)`, *optional*): Labels for computing
LETTER-LC-Rec/modeling_letter.py:48
Methodforward
(self, x, label, idx, use_sk=True)
RQ-VAE/models/vq.py:157
Methodforward
(self, x, labels, use_sk=True)
RQ-VAE/models/rq.py:43
Methodforward
(self, input_feature)
RQ-VAE/models/layers.py:40
Methodforward
(self, x, labels, use_sk=True)
RQ-VAE/models/rqvae.py:68
Methodforward
r"""
LETTER-TIGER/modeling_letter.py:51
Methodget_all_items_v2
(self)
LETTER-TIGER/data.py:67
Methodget_codebook
(self)
RQ-VAE/models/vq.py:32
Methodget_codebook
(self)
RQ-VAE/models/rq.py:27
Methodget_codebook_entry
(self, indices, shape=None)
RQ-VAE/models/vq.py:35
Functionget_local_time
()
LETTER-LC-Rec/utils.py:121
Functionget_local_time
()
LETTER-TIGER/utils.py:107
Methodget_prefix_allowed_tokens_fn
(self, tokenizer)
LETTER-TIGER/data.py:76
Functionget_res_batch
(model_name, prompt_list, max_tokens, api_info)
data_process/utils.py:16
Functionget_topk_ranking_results
(predictions, targets, k, all_items=None)
LETTER-TIGER/evaluate.py:33
Methodinit_weights
(self, module)
RQ-VAE/models/layers.py:33
Functionkmeans
( samples, num_clusters, num_iters = 10, )
RQ-VAE/models/layers.py:67
Methodload_from_dict
(trie_dict)
LETTER-TIGER/generation_trie.py:33
Functionload_json
(file)
LETTER-LC-Rec/utils.py:192
Functionload_json
(file)
LETTER-TIGER/utils.py:169
Functionload_pickle
(filename)
data_process/utils.py:125
Functionmake_inters_in_order
(inters)
data_process/utils.py:130
Functionprefix_allowed_tokens
(batch_id, sentence)
LETTER-TIGER/generation_trie.py:92
Functionprefix_allowed_tokens
(batch_id, sentence)
LETTER-TIGER/utils.py:162
Methodprefix_allowed_tokens_fn
(batch_id, sentence)
LETTER-LC-Rec/data.py:81
Methodprefix_allowed_tokens_fn
(batch_id, sentence)
LETTER-TIGER/data.py:91
Methodset_prompt
(self, prompt_id)
LETTER-LC-Rec/data.py:411
Methodset_prompt
(self, prompt_id)
LETTER-LC-Rec/data.py:620
← previousnext →101–200 of 205, ranked by callers