Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/autoliuweijie/K-BERT
/ types & classes
Types & classes
52 in github.com/autoliuweijie/K-BERT
⨍
Functions
198
◇
Types & classes
52
↓ 5 callers
Class
LayerNorm
uer/layers/layer_norm.py:6
↓ 4 callers
Class
Vocab
uer/utils/vocab.py:16
↓ 3 callers
Class
BertAdam
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 callers
Class
MultiHeadedAttention
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 callers
Class
KnowledgeGraph
spo_files - list of Path of *.spo files, or default kg name. e.g., ['HowNet']
brain/knowgraph.py:11
↓ 2 callers
Class
TransformerLayer
Transformer layer mainly consists of two parts: multi-headed self-attention and feed forward layer.
uer/layers/transformer.py:8
↓ 1 callers
Class
BasicTokenizer
Runs basic tokenization (punctuation splitting, lower casing, etc.).
uer/utils/tokenizer.py:108
↓ 1 callers
Class
BertClassifier
run_kbert_cls.py:25
↓ 1 callers
Class
BertEmbedding
BERT embedding consists of three parts: word embedding, position embedding, and segment embedding.
uer/layers/embeddings.py:7
↓ 1 callers
Class
BertTagger
run_kbert_ner.py:22
↓ 1 callers
Class
Model
BertModel consists of three parts: - embedding: token embedding, position embedding, segment embedding - encoder: multi-layer tra
uer/models/model.py:8
↓ 1 callers
Class
PositionwiseFeedForward
Feed Forward Layer
uer/layers/position_ffn.py:6
↓ 1 callers
Class
WordpieceTokenizer
Runs WordPiece tokenization.
uer/utils/tokenizer.py:224
Class
AttnEncoder
BERT encoder exploits 12 or 24 transformer layers to extract features.
uer/encoders/attn_encoder.py:7
Class
AvgSubencoder
uer/subencoders/avg_subencoder.py:6
Class
BertDataLoader
uer/utils/data.py:268
Class
BertDataset
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
Class
BertEncoder
BERT encoder exploits 12 or 24 transformer layers to extract features.
uer/encoders/bert_encoder.py:9
Class
BertModel
BertModel consists of three parts: - embedding: token embedding, position embedding, segment embedding - encoder: multi-layer tra
uer/models/bert_model.py:5
Class
BertTarget
BERT exploits masked language modeling (MLM) and next sentence prediction (NSP) for pretraining.
uer/targets/bert_target.py:9
Class
BertTokenizer
Runs end-to-end tokenization: punctuation splitting + wordpiece
uer/utils/tokenizer.py:48
Class
BilmDataLoader
uer/utils/data.py:567
Class
BilmDataset
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
Class
BilmTarget
uer/targets/bilm_target.py:10
Class
BilstmEncoder
uer/encoders/birnn_encoder.py:6
Class
CharTokenizer
uer/utils/tokenizer.py:18
Class
ClsDataLoader
uer/utils/data.py:739
Class
ClsDataset
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
Class
ClsTarget
uer/targets/cls_target.py:9
Class
CnnEncoder
uer/encoders/cnn_encoder.py:6
Class
CnnSubencoder
uer/subencoders/cnn_subencoder.py:6
Class
CrnnEncoder
uer/encoders/mixed_encoder.py:53
Class
GatedcnnEncoder
uer/encoders/cnn_encoder.py:40
Class
GptEncoder
BERT encoder exploits 12 or 24 transformer layers to extract features.
uer/encoders/gpt_encoder.py:10
Class
GruEncoder
uer/encoders/rnn_encoder.py:43
Class
LmDataLoader
uer/utils/data.py:418
Class
LmDataset
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
Class
LmTarget
uer/targets/lm_target.py:9
Class
LstmEncoder
uer/encoders/rnn_encoder.py:6
Class
LstmSubencoder
uer/subencoders/rnn_subencoder.py:5
Class
MlmDataLoader
uer/utils/data.py:880
Class
MlmDataset
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
Class
MlmTarget
BERT exploits masked language modeling (MLM) and next sentence prediction (NSP) for pretraining.
uer/targets/mlm_target.py:9
Class
NspDataLoader
uer/utils/data.py:1116
Class
NspDataset
uer/utils/data.py:943
Class
NspTarget
uer/targets/nsp_target.py:9
Class
RcnnEncoder
uer/encoders/mixed_encoder.py:6
Class
S2sDataLoader
uer/utils/data.py:1261
Class
S2sDataset
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
Class
S2sTarget
uer/targets/s2s_target.py:9
Class
SpaceTokenizer
uer/utils/tokenizer.py:27
Class
Tokenizer
uer/utils/tokenizer.py:9