资源分类
》 资源 》算法》bert_classification

bert_classification

2020-03-10 | |  29 |   0 |   0

利用bert预训练的中文语言模型进行文本分类

训练脚本 train.sh

批量测试脚本 predict.sh

利用模型进行单条语句测试 intent.py 注：修改single_predict.py中 get_test_examples,get_labels方法 max_seq_length，需run_classifier.py一致

chinese_L-12_H-768_A-12为预训练的相关模型和词典

data文件夹中为训练语料，验证语料，测试语料

注意： 当语料中的分类种类发生变化时，run_classifier.py文件的SimProcessor中的get_labels方法也要随之变化

参数说明： max_seq_length sentence的最大长度（字） train_batch_size batch_size的大小


上一篇： BERT-for-RRC-ABSA

下一篇：chinese-bert-similarity

用户评价
全部评价

热门资源

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • shih-styletransfer

    shih-styletransfer Code from Style Transfer ...