使用bert预训练模型获取文本的嵌入向量表示

示例代码:

    os.system("python3 extract_features.py \
	  --input_file=input.txt \
	  --output_file=output.json \
	  --vocab_file=uncased_L-12_H-768_A-12/vocab.txt \
	  --bert_config_file=uncased_L-12_H-768_A-12/bert_config.json \
	  --init_checkpoint=uncased_L-12_H-768_A-12/bert_model.ckpt \
	  --layers=-1 \
	  --max_seq_length=256 \
	  --batch_size=8")

你可能感兴趣的:(AI)