-
Tensorflow 实现的半监督序列学习:transfer-learning-text-tf (https
资源介绍
使用Tensorflow进行文本分类的转移学习
Tensorflow实现 。
使用自动编码器或语言模型作为预训练模型来初始化LSTM文本分类模型。
SA-LSTM :使用自动编码器作为预训练模型。
LM-LSTM :使用语言模型作为预训练模型。
要求
的Python 3
张量流
点安装-r requirements.txt
用法
DBpedia数据集用于预训练和训练。
训练前自动编码器或语言模型
$ python pre_train.py --model=""
(<模型>:auto_encoder |语言模型)
训练LSTM文本分类模型
$ python train.py --pre_trained=""
(<模型>:无|自动编码器|语言模型)
实验结果
橙色线:LSTM
蓝线:SA-LSTM
红线:LM-LSTM
失利
准确性