MCPcopy Create free account

hub / github.com/autoliuweijie/K-BERT / types & classes

Types & classes52 in github.com/autoliuweijie/K-BERT

↓ 5 callersClassLayerNorm
uer/layers/layer_norm.py:6
↓ 4 callersClassVocab
uer/utils/vocab.py:16
↓ 3 callersClassBertAdam
Implements BERT version of Adam algorithm with weight decay fix. Params: lr: learning rate warmup: portion of t_total for the
uer/utils/optimizers.py:35
↓ 3 callersClassMultiHeadedAttention
Each head is a self-attention operation. self-attention refers to https://arxiv.org/pdf/1706.03762.pdf
uer/layers/multi_headed_attn.py:6
↓ 2 callersClassKnowledgeGraph
spo_files - list of Path of *.spo files, or default kg name. e.g., ['HowNet']
brain/knowgraph.py:11
↓ 2 callersClassTransformerLayer
Transformer layer mainly consists of two parts: multi-headed self-attention and feed forward layer.
uer/layers/transformer.py:8
↓ 1 callersClassBasicTokenizer
Runs basic tokenization (punctuation splitting, lower casing, etc.).
uer/utils/tokenizer.py:108
↓ 1 callersClassBertClassifier
run_kbert_cls.py:25
↓ 1 callersClassBertEmbedding
BERT embedding consists of three parts: word embedding, position embedding, and segment embedding.
uer/layers/embeddings.py:7
↓ 1 callersClassBertTagger
run_kbert_ner.py:22
↓ 1 callersClassModel
BertModel consists of three parts: - embedding: token embedding, position embedding, segment embedding - encoder: multi-layer tra
uer/models/model.py:8
↓ 1 callersClassPositionwiseFeedForward
Feed Forward Layer
uer/layers/position_ffn.py:6
↓ 1 callersClassWordpieceTokenizer
Runs WordPiece tokenization.
uer/utils/tokenizer.py:224
ClassAttnEncoder
BERT encoder exploits 12 or 24 transformer layers to extract features.
uer/encoders/attn_encoder.py:7
ClassAvgSubencoder
uer/subencoders/avg_subencoder.py:6
ClassBertDataLoader
uer/utils/data.py:268
ClassBertDataset
Construct dataset for MLM and NSP tasks from the given corpus. Each document consists of multiple sentences, and each sentence occupies
uer/utils/data.py:68
ClassBertEncoder
BERT encoder exploits 12 or 24 transformer layers to extract features.
uer/encoders/bert_encoder.py:9
ClassBertModel
BertModel consists of three parts: - embedding: token embedding, position embedding, segment embedding - encoder: multi-layer tra
uer/models/bert_model.py:5
ClassBertTarget
BERT exploits masked language modeling (MLM) and next sentence prediction (NSP) for pretraining.
uer/targets/bert_target.py:9
ClassBertTokenizer
Runs end-to-end tokenization: punctuation splitting + wordpiece
uer/utils/tokenizer.py:48
ClassBilmDataLoader
uer/utils/data.py:567
ClassBilmDataset
Construct dataset for MLM and NSP tasks from the given corpus. Each document consists of multiple sentences, and each sentence occupies
uer/utils/data.py:481
ClassBilmTarget
uer/targets/bilm_target.py:10
ClassBilstmEncoder
uer/encoders/birnn_encoder.py:6
ClassCharTokenizer
uer/utils/tokenizer.py:18
ClassClsDataLoader
uer/utils/data.py:739
ClassClsDataset
Construct dataset for MLM and NSP tasks from the given corpus. Each document consists of multiple sentences, and each sentence occupies
uer/utils/data.py:633
ClassClsTarget
uer/targets/cls_target.py:9
ClassCnnEncoder
uer/encoders/cnn_encoder.py:6
ClassCnnSubencoder
uer/subencoders/cnn_subencoder.py:6
ClassCrnnEncoder
uer/encoders/mixed_encoder.py:53
ClassGatedcnnEncoder
uer/encoders/cnn_encoder.py:40
ClassGptEncoder
BERT encoder exploits 12 or 24 transformer layers to extract features.
uer/encoders/gpt_encoder.py:10
ClassGruEncoder
uer/encoders/rnn_encoder.py:43
ClassLmDataLoader
uer/utils/data.py:418
ClassLmDataset
Construct dataset for MLM and NSP tasks from the given corpus. Each document consists of multiple sentences, and each sentence occupies
uer/utils/data.py:334
ClassLmTarget
uer/targets/lm_target.py:9
ClassLstmEncoder
uer/encoders/rnn_encoder.py:6
ClassLstmSubencoder
uer/subencoders/rnn_subencoder.py:5
ClassMlmDataLoader
uer/utils/data.py:880
ClassMlmDataset
Construct dataset for MLM and NSP tasks from the given corpus. Each document consists of multiple sentences, and each sentence occupies
uer/utils/data.py:802
ClassMlmTarget
BERT exploits masked language modeling (MLM) and next sentence prediction (NSP) for pretraining.
uer/targets/mlm_target.py:9
ClassNspDataLoader
uer/utils/data.py:1116
ClassNspDataset
uer/utils/data.py:943
ClassNspTarget
uer/targets/nsp_target.py:9
ClassRcnnEncoder
uer/encoders/mixed_encoder.py:6
ClassS2sDataLoader
uer/utils/data.py:1261
ClassS2sDataset
Construct dataset for MLM and NSP tasks from the given corpus. Each document consists of multiple sentences, and each sentence occupies
uer/utils/data.py:1178
ClassS2sTarget
uer/targets/s2s_target.py:9
ClassSpaceTokenizer
uer/utils/tokenizer.py:27
ClassTokenizer
uer/utils/tokenizer.py:9