MCPcopy Create free account

hub / github.com/ProHiryu/bert-chinese-ner / functions

Functions31 in github.com/ProHiryu/bert-chinese-ner

⨍ Functions 31◇ Types & classes 4
↓ 6 callersFunctionmetrics_from_confusion_matrix
Precision, Recall and F1 from the confusion matrix Parameters ---------- cm : tf.Tensor of type tf.int32, of shape (num_classes, num_cl
tf_metrics.py:168
↓ 6 callersFunctionsafe_div
Safe division, return 0 if denominator is 0
tf_metrics.py:137
↓ 4 callersFunctioncreate_int_feature
(values)
BERT_NER.py:305
↓ 3 callersMethod_create_example
(self, lines, set_type)
BERT_NER.py:194
↓ 3 callersMethod_read_data
Reads a BIO data.
BERT_NER.py:147
↓ 3 callersFunctionfile_based_input_fn_builder
(input_file, seq_length, is_training, drop_remainder)
BERT_NER.py:319
↓ 3 callersFunctionfiled_based_convert_examples_to_features
( examples, label_list, max_seq_length, tokenizer, output_file,mode=None )
BERT_NER.py:290
↓ 2 callersFunctionpr_re_fbeta
Uses a confusion matrix to compute precision, recall and fbeta
tf_metrics.py:145
↓ 1 callersFunction_decode_record
(record, name_to_features)
BERT_NER.py:329
↓ 1 callersFunctionconvert_single_example
(ex_index, example, label_map, max_seq_length, tokenizer,mode)
BERT_NER.py:213
↓ 1 callersFunctioncreate_model
(bert_config, is_training, input_ids, input_mask, segment_ids, labels, num_labels, use_one_h
BERT_NER.py:353
↓ 1 callersFunctionfbeta
Multi-class fbeta metric for Tensorflow Parameters ---------- labels : Tensor of tf.int32 or tf.int64 The true labels pre
tf_metrics.py:97
↓ 1 callersMethodget_dev_examples
(self, data_dir)
BERT_NER.py:179
↓ 1 callersMethodget_labels
(self)
BERT_NER.py:189
↓ 1 callersMethodget_test_examples
(self,data_dir)
BERT_NER.py:184
↓ 1 callersMethodget_train_examples
(self, data_dir)
BERT_NER.py:174
↓ 1 callersFunctionmodel_fn_builder
(bert_config, num_labels, init_checkpoint, learning_rate, num_train_steps, num_warmup_st
BERT_NER.py:395
↓ 1 callersFunctionwrite_tokens
(tokens,mode)
BERT_NER.py:204
Method__init__
Constructs a InputExample. Args: guid: Unique id for the example. text_a: string. The untokenized text of the first s
BERT_NER.py:105
Method__init__
(self, input_ids, input_mask, segment_ids, label_ids,)
BERT_NER.py:123
Functionf1
(labels, predictions, num_classes, pos_indices=None, weights=None, average='micro')
tf_metrics.py:91
Methodget_dev_examples
Gets a collection of `InputExample`s for the dev set.
BERT_NER.py:138
Methodget_labels
Gets the list of labels for this data set.
BERT_NER.py:142
Methodget_train_examples
Gets a collection of `InputExample`s for the train set.
BERT_NER.py:134
Functioninput_fn
(params)
BERT_NER.py:338
Functionmain
(_)
BERT_NER.py:471
Functionmetric_fn
(per_example_loss, label_ids, logits)
BERT_NER.py:443
Functionmodel_fn
(features, labels, mode, params)
BERT_NER.py:398
Functionprecision
Multi-class precision metric for Tensorflow Parameters ---------- labels : Tensor of tf.int32 or tf.int64 The true labels
tf_metrics.py:15
Functionrecall
Multi-class recall metric for Tensorflow Parameters ---------- labels : Tensor of tf.int32 or tf.int64 The true labels pr
tf_metrics.py:53
Functiontpu_scaffold
()
BERT_NER.py:418