MCPcopy Create free account

hub / github.com/RingBDStack/GDAP / functions

Functions161 in github.com/RingBDStack/GDAP

↓ 7 callersFunctionfreeze_params
Set requires_grad=False for each of model.parameters()
seq2seq/utils.py:544
↓ 7 callersFunctionlmap
list(map(f, x))
seq2seq/utils.py:68
↓ 6 callersFunctionclean_text
(tree_str)
extraction/predict_parser/target_predict_parser.py:56
↓ 6 callersMethodread_from_file
(filename)
extraction/event_schema.py:15
↓ 5 callersFunctionget_label_name_tree
(label_name_list, tokenizer, end_symbol='<end>')
extraction/label_tree.py:26
↓ 5 callersFunctionread_file
(file_name)
evaluation.py:12
↓ 4 callersMethodcount_instance
(self, gold_list, pred_list, verbose=False, text=None)
evaluation.py:165
↓ 4 callersMethoddecode
:param gold_list: :param pred_list: :param text_list: :param raw_list: :return: dict:
extraction/predict_parser/predict_parser.py:15
↓ 4 callersFunctiongenerated_search_src_sequence
(generated, src_sequence, end_sequence_search_tokens=None)
extraction/extract_constraint.py:38
↓ 4 callersFunctiontrim_batch
Remove columns that are populated exclusively by pad_token_id
seq2seq/utils.py:107
↓ 3 callersMethod__init__
(self, tokenizer, source_prefix)
extraction/extract_constraint.py:77
↓ 3 callersFunctionadd_space
add space between special token :param text: :return:
extraction/predict_parser/target_predict_parser.py:24
↓ 3 callersFunctionconvert_dyiepp_event
(output_folder='data/text2target/ace2005_event', type_format='subtype', ignore_noneve
data_convert/convert_text_to_target.py:93
↓ 3 callersMethodcount_multi_event_role_in_instance
(instance, counter)
extraction/predict_parser/predict_parser.py:33
↓ 3 callersFunctionfind_bracket_position
(generated_text, _type_start, _type_end)
extraction/extract_constraint.py:30
↓ 3 callersFunctiongrad_status
(model: nn.Module)
seq2seq/utils.py:569
↓ 3 callersMethodsafe_div
(a, b)
evaluation.py:146
↓ 3 callersMethodsafe_div
(a, b)
extraction/predict_parser/predict_parser.py:54
↓ 2 callersFunctionadd_newline_to_end_of_each_sentence
This was added to get rougeLsum scores matching published rougeL scores for BART and PEGASUS.
seq2seq/sentence_splitter.py:35
↓ 2 callersMethodannotate_span
src: source text traget: ((ET)) 只对ET进行生成, ((ET)(ET)...) :param tokens: US President George W. Bush told Canadia
data_convert/format/text2target.py:132
↓ 2 callersFunctionclean_str
(x_str)
run_seq2seq.py:624
↓ 2 callersMethodcompute_f1
(self, prefix='')
evaluation.py:152
↓ 2 callersMethodcompute_f1
(self, prefix='')
extraction/predict_parser/predict_parser.py:60
↓ 2 callersMethodcount_instance
(self, gold_list, pred_list, verbose=False)
extraction/predict_parser/predict_parser.py:73
↓ 2 callersFunctiondecode_pred
(pred: EvalPrediction)
seq2seq/utils.py:82
↓ 2 callersFunctiondecode_tree_str
(sequences: Union[List[int], List[List[int]], "np.ndarray", "torch.Tensor"], tokenizer: Pr
seq2seq/constrained_seq2seq.py:36
↓ 2 callersMethodencode_line
Only used by LegacyDataset
seq2seq/utils.py:224
↓ 2 callersFunctionfind_bracket_num
Count Bracket Number, 0 indicate num_left = num_right :param tree_str: :return:
extraction/predict_parser/target_predict_parser.py:35
↓ 2 callersMethodget_char_lens
(data_file)
seq2seq/utils.py:158
↓ 2 callersMethodget_event_list
(self, span_str, text=None, et=None, rt=None)
extraction/predict_parser/target_predict_parser.py:192
↓ 2 callersMethodget_event_list
(self, span_str, text=None, et=None, rt=None)
extraction/predict_parser/target_predict_parser.py:312
↓ 2 callersMethodget_event_list
(self, span_str, text=None, et=None, rt=None)
extraction/predict_parser/target_predict_parser.py:414
↓ 2 callersFunctionget_extract_metrics
(pred_lns: List[str], tgt_lns: List[str], label_constraint: EventSchema, decoding_format='tree')
extraction/extraction_metrics.py:45
↓ 2 callersFunctionget_str_from_tokens
(tokens, sentence, separator=' ')
data_convert/format/text2target.py:10
↓ 2 callersFunctionmain
()
run_seq2seq.py:279
↓ 2 callersFunctionmatch_sublist
:param the_list: [1, 2, 3, 4, 5, 6, 1, 2, 4, 5] :param to_match: [1, 2] :return: [(0, 1), (6, 7)]
evaluation.py:80
↓ 2 callersFunctionread_file
(file_name)
convert_et_result.py:21
↓ 2 callersFunctionsortish_sampler_indices
Go through the text data by order of src length with a bit of randomness. From fastai repo.
seq2seq/utils.py:346
↓ 1 callersMethod__init__
(self, data, batch_size, shuffle=True)
seq2seq/utils.py:336
↓ 1 callersMethod_encode
(self, batch)
seq2seq/utils.py:320
↓ 1 callersFunctioncalculate_bleu
Uses sacrebleu's corpus_bleu implementation.
seq2seq/utils.py:73
↓ 1 callersFunctioncalculate_rouge
Calculate rouge using rouge_scorer package. Args: pred_lns: list of summaries generated by model tgt_lns: list of groundtruth summ
seq2seq/utils.py:496
↓ 1 callersFunctioncheck_output
(filename, line_num=2)
data_convert/utils.py:18
↓ 1 callersMethodcheck_state
(self, tgt_generated)
extraction/extract_constraint.py:122
↓ 1 callersMethodcheck_state
(self, tgt_generated)
extraction/extract_constraint.py:263
↓ 1 callersMethodcheck_state
(self, tgt_generated)
extraction/extract_constraint.py:383
↓ 1 callersMethodconstraint_decoding
(self, src_sentence, tgt_generated)
extraction/extract_constraint.py:86
↓ 1 callersFunctionconvert_file_tuple
(file_tuple, data_class=Event, target_class=ETRTText2Role, output_folder='data/text2tar
data_convert/convert_text_to_target.py:13
↓ 1 callersFunctiondata_counter_to_table
(data_counter)
data_convert/utils.py:23
↓ 1 callersFunctiondecode_pred
(pred: EvalPrediction)
seq2seq/constrained_seq2seq.py:53
↓ 1 callersFunctionet_text2role
(schema, et_list, text)
convert_et_result.py:25
↓ 1 callersFunctionet_text2tri
(et_list, text)
convert_et_result.py:44
↓ 1 callersFunctioneval_pred
(predict_parser, gold_list, pred_list, text_list=None, raw_list=None)
extraction/extraction_metrics.py:18
↓ 1 callersFunctionextract_rouge_mid_statistics
(dct)
seq2seq/utils.py:488
↓ 1 callersMethodgenerate_sentence
(self, type_format='subtype')
data_convert/task_format/event_extraction.py:88
↓ 1 callersFunctiongenerate_sentence_dyiepp
(filename, type_format='subtype')
evaluation.py:16
↓ 1 callersFunctiongenerate_sentence_text2target
(filename, pred_reader)
evaluation.py:57
↓ 1 callersFunctionget_constraint_decoder
(tokenizer, type_schema, decoding_schema, source_prefix=None)
extraction/extract_constraint.py:59
↓ 1 callersFunctionget_git_info
()
seq2seq/utils.py:466
↓ 1 callersFunctionget_predict_parser
(format_name)
extraction/extraction_metrics.py:14
↓ 1 callersFunctionget_schema
(event)
data_convert/utils.py:31
↓ 1 callersMethodget_state_valid_tokens
(self, src_sentence: List[str], tgt_generated: List[str])
extraction/extract_constraint.py:83
↓ 1 callersFunctionkey_fn
(i)
seq2seq/utils.py:351
↓ 1 callersFunctionlist_dictionary
(d, n_tab=-1)
extraction/label_tree.py:6
↓ 1 callersFunctionmain
()
evaluation.py:183
↓ 1 callersFunctionmain
()
convert_dyiepp_to_sentence.py:9
↓ 1 callersFunctionmain
()
seq2seq/constrained_seq2seq.py:204
↓ 1 callersMethodmake_sortish_sampler
(self, batch_size, distributed=False, shuffle=True, **kwargs)
seq2seq/utils.py:166
↓ 1 callersFunctionmatch_sublist
:param the_list: [1, 2, 3, 4, 5, 6, 1, 2, 4, 5] :param to_match: [1, 2] :return: [(0, 1), (6, 7)]
extraction/extract_constraint.py:14
↓ 1 callersFunctionoutput_schema
(event_schema_set, output_file)
data_convert/utils.py:38
↓ 1 callersFunctionpickle_load
pickle.load(path)
seq2seq/utils.py:434
↓ 1 callersFunctionprint_tree
(tree)
extraction/label_tree.py:22
↓ 1 callersFunctionread_file
(filename)
data_convert/utils.py:10
↓ 1 callersFunctionsave_json
(content, path, indent=4, **json_dump_kwargs)
seq2seq/utils.py:456
↓ 1 callersMethodsearch_prefix_tree
Generate Text Span :param generated: :param prefix_tree: :param src_sentence: :param end_sequence_search_toke
extraction/extract_constraint.py:292
FunctionDyIEPP_ace2005_file_tuple
(output_folder)
data_convert/task_format/event_extraction.py:115
Method__call__
(self, batch)
seq2seq/utils.py:290
Method__call__
(self, model_output, labels)
seq2seq/label_smoother_sum.py:22
Method__getitem__
(self, item)
seq2seq/utils.py:197
Method__getitem__
Call tokenizer on src and tgt_lines
seq2seq/utils.py:205
Method__getitem__
(self, index)
seq2seq/utils.py:253
Method__init__
(self)
evaluation.py:140
Method__init__
(self, tokenizer, type_schema, *args, **kwargs)
extraction/extract_constraint.py:112
Method__init__
(self, tokenizer, type_schema, *args, **kwargs)
extraction/extract_constraint.py:254
Method__init__
(self, tokenizer, type_schema, *args, **kwargs)
extraction/extract_constraint.py:373
Method__init__
(self, type_list, role_list, type_role_dict)
extraction/event_schema.py:9
Method__init__
(self, label_name_list, tokenizer, end_symbol='<end>')
extraction/label_tree.py:47
Method__init__
(self, schema)
extraction/predict_parser/predict_parser.py:11
Method__init__
(self)
extraction/predict_parser/predict_parser.py:48
Method__init__
( self, tokenizer, data_dir, max_source_length, ma
seq2seq/utils.py:121
Method__init__
(self, tokenizer, data_args, tpu_num_cores=None)
seq2seq/utils.py:276
Method__init__
(self, dataset, batch_size, num_replicas=None, rank=None, add_extra_examples=True, shuffle=True)
seq2seq/utils.py:370
Method__init__
(self, decoding_type_schema=None, decoding_format='tree', source_prefix=None, *args, **kwargs)
seq2seq/constrained_seq2seq.py:84
Method__init__
(self, doc_json)
data_convert/task_format/event_extraction.py:9
Method__init__
(self, doc_json)
data_convert/task_format/event_extraction.py:78
Method__iter__
(self)
seq2seq/utils.py:342
Method__iter__
(self)
seq2seq/utils.py:393
Method__len__
(self)
seq2seq/utils.py:154
Method__len__
(self)
seq2seq/utils.py:339
Method__len__
(self)
seq2seq/utils.py:413
next →1–100 of 161, ranked by callers