/text_classification

关于文本分类的许多方法,主要涉及到TextCNN,TextRNN, LEAM, Transformer,Attention, fasttext, HAN等

Primary LanguagePython

data包中train_sentiment.txt为训练语料 train_fasttext.txt为fasttext的训练语料

model包下为文本分类的模型

save_model包下为保存的模型

config为各个模型对应的相应参数配置

train.sh 指定模型进行训练

infer.sh 指定模型和语句进行预测