Mit diesem Repository können Sie die Multi-Label-Klassifizierung mit Bert trainieren.
Bert Bert für die Online -Vorhersage einsetzen.
Sie finden auch ein kurzes Tutorial für die Verwendung von Bert mit Chinesisch: Bert Short Chinese Tutorial
Sie können von AI Challenger Einführung in die feine Getreidestimmung finden
Fügen Sie hier etwas hinzu.

Weitere Informationen finden Sie unter Modell/BERT_CNN_FINE_GRAIN_MODEL.PY
| Modell | Textcnn (No-Presen) | Textcnn (vorab-finetuning) | Bert (Base_Model_zh) | Bert (Base_model_zh, vor dem Corpus) |
|---|---|---|---|---|
| F1 -Punktzahl | 0,678 | 0,685 | Fügen Sie hier eine Nummer hinzu | Fügen Sie hier eine Nummer hinzu |
Hinweis: Die F1 -Punktzahl wird im Validierungssatz gemeldet

export BERT_BASE_DIR=BERT_BASE_DIR/chinese_L-12_H-768_A-12
export TEXT_DIR=TEXT_DIR
nohup python run_classifier_multi_labels_bert.py
--task_name=sentiment_analysis
--do_train=true
--do_eval=true
--data_dir=$TEXT_DIR
--vocab_file=$BERT_BASE_DIR/vocab.txt
--bert_config_file=$BERT_BASE_DIR/bert_config.json
--init_checkpoint=$BERT_BASE_DIR/bert_model.ckpt
--max_seq_length=512
--train_batch_size=4
--learning_rate=2e-5
--num_train_epochs=3
--output_dir=./checkpoint_bert &
1.First, müssen Sie ein vorgebildetes Modell von Google herunterladen und in einen Ordner (egbert_base_dir) einstellen.
chinese_L-12_H-768_A-12 from <a href='https://storage.googleapis.com/bert_models/2018_11_03/chinese_L-12_H-768_A-12.zip'>bert</a>
2.SEKONDLIS MÜSSEN Sie Schulungsdaten (z. B. Train.TSV) und Validierungsdaten (z.
folder(e.g.TEXT_DIR ). you can also download data from here <a href='https://pan.baidu.com/s/1ZS4dAdOIAe3DaHiwCDrLKw'>data to train bert for AI challenger-Sentiment Analysis</a>.
it contains processed data you can run for both fine-tuning on sentiment analysis and pre-train with Bert.
it is generated by following this notebook step by step:
preprocess_char.ipynb
you can generate data by yourself as long as data format is compatible with
processor SentimentAnalysisFineGrainProcessor(alias as sentiment_analysis);
data format: label1,label2,label3t here is sentence or sentencest
it only contains two columns, the first one is target(one or multi-labels), the second one is input strings.
no need to tokenized.
sample:"0_1,1_-2,2_-2,3_-2,4_1,5_-2,6_-2,7_-2,8_1,9_1,10_-2,11_-2,12_-2,13_-2,14_-2,15_1,16_-2,17_-2,18_0,19_-2 浦东五莲路站,老饭店福瑞轩属于上海的本帮菜,交通方便,最近又重新装修,来拨草了,饭店活动满188元送50元钱,环境干净,简单。朋友提前一天来预订包房也没有订到,只有大堂,五点半到店基本上每个台子都客满了,都是附近居民,每道冷菜量都比以前小,味道还可以,热菜烤茄子,炒河虾仁,脆皮鸭,照牌鸡,小牛排,手撕腊味花菜等每道菜都很入味好吃,会员价划算,服务员人手太少,服务态度好,要能团购更好。可以用支付宝方便"
check sample data in ./BERT_BASE_DIR folder
for more detail, check create_model and SentimentAnalysisFineGrainProcessor from run_classifier.py
Rawdaten generieren: [Fügen Sie hier etwas hinzu]
Nehmen wir sicher, jede Zeile ist ein Satz. Zwischen jedem Dokument befindet sich eine leere Zeile.
Sie können generierte Daten aus der ZIP -Datei finden.
use write_pre_train_doc() from preprocess_char.ipynb
Generieren Sie Daten für die Stufe vor dem Trainer mit:
export BERT_BASE_DIR=./BERT_BASE_DIR/chinese_L-12_H-768_A-12
nohup python create_pretraining_data.py
--input_file=./PRE_TRAIN_DIR/bert_*_pretrain.txt
--output_file=./PRE_TRAIN_DIR/tf_examples.tfrecord
--vocab_file=$BERT_BASE_DIR/vocab.txt
--do_lower_case=True
--max_seq_length=512
--max_predictions_per_seq=60
--masked_lm_prob=0.15
--random_seed=12345
--dupe_factor=5 nohup_pre.out &
Pre-Training-Modell mit generierten Daten:
python run_pretring.py
Feinabstimmung
python run_classifier.py
Laden Sie die Cache -Datei mit der Sentiment -Analyse herunter (Token sind in Wortebene)
trainieren Sie das Modell:
Python Train_cnn_fine_grain.py
cache file of TextCNN model was generate by following steps from preprocess_word.ipynb.
it contains everything you need to run TextCNN.
it include: processed train/validation/test set; vocabulary of word; a dict map label to index.
take train_valid_test_vocab_cache.pik and put it under folder of preprocess_word/
raw data are also included in this zip file.
Pre-Training TextCnn mit maskiertem Sprachmodell
Python Train_cnn_lm.py
Feinabstimmung für Textcnn
Python Train_cnn_fine_grain.py
with session and feed style you can easily deploy BERT.
Online -Vorhersage mit Bert, überprüfen Sie mehr von hier aus
Bidirektionale Encoder -Repräsentationen von Transformatoren für das Sprachverständnis
Google-Research/Bert
Pengshuang/ai-comp
AI Challenger 2018
Faltungsnetzwerke für die Satzklassifizierung