MCPcopy Create free account

hub / github.com/chenchens190009/KG-S2S / functions

Functions95 in github.com/chenchens190009/KG-S2S

⨍ Functions 95◇ Types & classes 21
↓ 25 callersMethodwrite_file
(self, filename, out_folder=None, dataset=None, sort_key=None, func=lambda x: x)
script/base.py:35
↓ 22 callersMethodread_file
(self, filename, in_folder=None, dataset=None)
script/base.py:23
↓ 15 callersFunctionbatchify
(output_dict, key, padding_value=None, return_list=False)
helper.py:126
↓ 5 callersMethodcreate_out_folder
(self)
script/base.py:20
↓ 4 callersMethodread_triples
(self, filename, in_folder=None, dataset=None)
script/base.py:57
↓ 4 callersMethodwrite_triples
(self, filename, triples, out_folder=None, dataset=None)
script/base.py:69
↓ 3 callersMethodget_soft_prompt_input_embed
(self, src_ids, src_mask, ent_rel, input_index, soft_prompt_index)
models/model.py:286
↓ 3 callersFunctionread
(configs, dataset_path, dataset, filename)
helper.py:19
↓ 3 callersFunctionread_file
(configs, dataset_path, dataset, filename, mode='descrip')
helper.py:36
↓ 2 callersFunction_get_performance
(ranks, dataset)
helper.py:139
↓ 2 callersFunctionconstruct_prefix_trie
(ent_token_ids_in_trie)
helper.py:119
↓ 2 callersMethodconvert_time
(self, time)
script/process_icews14.py:93
↓ 2 callersFunctionget_ground_truth
(configs, triples)
helper.py:58
↓ 2 callersFunctionget_next_token_dict
(configs, ent_token_ids_in_trie, prefix_trie)
helper.py:72
↓ 2 callersFunctionget_num
(dataset_path, dataset, mode='entity')
helper.py:15
↓ 2 callersFunctionget_soft_prompt_pos
(configs, source_ids, target_ids, mode)
helper.py:94
↓ 1 callersMethodcreate_ent2descrip
(self, filename)
script/process_fb15k237.py:25
↓ 1 callersMethodcreate_ent2descrip
(self, filename)
script/process_fb15k237n.py:24
↓ 1 callersMethodcreate_ent2id
(self, filename)
script/process_fb15k237.py:37
↓ 1 callersMethodcreate_ent2id
(self, filename)
script/process_fb15k237n.py:36
↓ 1 callersMethodcreate_ent2id
(self, filename)
script/process_wn18rr.py:23
↓ 1 callersMethodcreate_ent2id_rel2id
(self)
script/process_nell.py:16
↓ 1 callersMethodcreate_ent2id_rel2id_entid2name_relid2name
(self)
script/process_icews14.py:27
↓ 1 callersMethodcreate_ent2name
(self, filename)
script/process_fb15k237.py:15
↓ 1 callersMethodcreate_ent2name
(self, filename)
script/process_fb15k237n.py:14
↓ 1 callersMethodcreate_ent2name_ent2descrip
(self, filename)
script/process_wn18rr.py:13
↓ 1 callersMethodcreate_entid2descrip
(self)
script/process_nell.py:69
↓ 1 callersMethodcreate_entid2descrip
(self, filename)
script/process_icews14.py:48
↓ 1 callersMethodcreate_entid2name
(self, filename)
script/process_nell.py:34
↓ 1 callersMethodcreate_rel2id
(self, filename)
script/process_fb15k237.py:48
↓ 1 callersMethodcreate_rel2id
(self, filename)
script/process_fb15k237n.py:47
↓ 1 callersMethodcreate_rel2id
(self, filename)
script/process_wn18rr.py:34
↓ 1 callersMethodcreate_rel2name
(self, filename)
script/process_nell.py:73
↓ 1 callersMethodcreate_typecons
(self, filename)
script/process_nell.py:80
↓ 1 callersMethoddecode
(self, src_ids, src_mask, batched_data)
models/model.py:156
↓ 1 callersFunctionget_performance
(model, tail_ranks, head_ranks)
helper.py:152
↓ 1 callersMethodlog_generation
(self, group_text, src_names, target_names, batch_idx, dataset_idx)
models/model.py:317
↓ 1 callersFunctionmain
()
main.py:17
↓ 1 callersMethodprocess_name
(self, name)
script/process_nell.py:43
↓ 1 callersMethodread_json_file
(self, filename, in_folder=None, dataset=None)
script/base.py:30
↓ 1 callersFunctionread_name
(configs, dataset_path, dataset)
helper.py:50
↓ 1 callersMethodread_triples
(self, filename, in_folder=None, dataset=None)
script/process_icews14.py:65
↓ 1 callersMethodvalidation_epoch_end
(self, outs)
models/model.py:342
↓ 1 callersMethodvalidation_step
(self, batched_data, batch_idx, dataset_idx)
models/model.py:102
↓ 1 callersMethodwrite_triples
(self, filename, triples, out_folder=None, dataset=None)
script/process_icews14.py:81
↓ 1 callersMethodwrite_typecons
(self, filename)
script/process_nell.py:91
Method__getitem__
(self, index)
data.py:24
Method__getitem__
(self, index)
data.py:109
Method__init__
(self, optimizer, warmup_updates, tot_updates, lr, end_lr, power, last_epoch=-1, verbose=False)
lr.py:5
Method__init__
(self, configs, tokenizer, train_triples, name_list_dict, prefix_trie_dict, ground_truth_dict)
data.py:9
Method__init__
(self, configs, tokenizer, test_triples, name_list_dict, prefix_trie_dict, ground_truth_dict, mode)
data.py:94
Method__init__
(self, configs, train_triples, valid_triples, test_triples, name_list_dict, prefix_trie_dict, ground_truth_dic
data.py:172
Method__init__
(self, in_folder, out_folder, dataset)
script/process_fb15k237.py:12
Method__init__
(self, in_folder, out_folder, in_dataset, out_dataset)
script/process_fb15k237n.py:10
Method__init__
(self, in_folder, out_folder, dataset)
script/base.py:6
Method__init__
(self, in_folder, out_folder, dataset)
script/process_nell.py:13
Method__init__
(self, in_folder, out_folder, dataset)
script/process_wn18rr.py:9
Method__init__
(self, in_folder, out_folder, dataset)
script/process_icews14.py:10
Method__init__
(self, configs, ground_truth_dict, name_list_dict, prefix_trie_dict=None)
models/model.py:14
Method__init__
(self, config)
models/modified_model/modified_T5.py:123
Method__len__
(self)
data.py:21
Method__len__
(self)
data.py:106
Method_extract
(generated_text)
models/model.py:157
Method_get_closed_form_lr
(self)
lr.py:28
Method_next_candidate
(configs, batch_idx, input_ids)
models/model.py:172
Method_reorder_cache
(self, past, beam_idx)
models/modified_model/modified_T5.py:346
Methodcollate_fn
(self, data)
data.py:78
Methodcollate_fn
(self, data)
data.py:157
Methodconfigure_optimizers
(self)
models/model.py:374
Methodconvert_ent_name
(name)
script/process_wn18rr.py:42
Methoddeparallelize
(self)
models/modified_model/modified_T5.py:162
Methodforward
( self, input_ids=None, attention_mask=None, decoder_input_ids=None, d
models/modified_model/modified_T5.py:192
Methodget_decoder
(self)
models/modified_model/modified_T5.py:189
Methodget_encoder
(self)
models/modified_model/modified_T5.py:186
Methodget_input_embeddings
(self)
models/modified_model/modified_T5.py:172
Methodget_lr
(self)
lr.py:13
Methodget_output_embeddings
(self)
models/modified_model/modified_T5.py:183
Methodmapper
(word)
script/process_nell.py:56
Methodon_test_end
(self, trainer, pl_module)
callbacks.py:29
Methodon_train_epoch_end
(self, trainer, pl_module)
callbacks.py:10
Methodon_train_epoch_start
(self, trainer, pl_module)
callbacks.py:6
Methodon_validation_end
(self, trainer, pl_module)
callbacks.py:25
Methodon_validation_start
(self, trainer, pl_module)
callbacks.py:20
Methodparallelize
(self, device_map=None)
models/modified_model/modified_T5.py:150
Methodprepare_data
(self)
data.py:188
Methodprepare_decoder_input_ids_from_labels
(self, labels: torch.Tensor)
models/modified_model/modified_T5.py:343
Methodprepare_inputs_for_generation
( self, input_ids, past=None, attention_mask=None, head_mask=None,
models/modified_model/modified_T5.py:315
Methodset_input_embeddings
(self, new_embeddings)
models/modified_model/modified_T5.py:175
Methodset_output_embeddings
(self, new_embeddings)
models/modified_model/modified_T5.py:180
Methodtest_dataloader
(self)
data.py:219
Methodtest_epoch_end
(self, outs)
models/model.py:371
Methodtest_step
(self, batched_data, batch_idx, dataset_idx)
models/model.py:368
Methodtrain_dataloader
(self)
data.py:195
Methodtraining_step
(self, batched_data, batch_idx)
models/model.py:46
Methodval_dataloader
(self)
data.py:204