/tensorflow-logistics-regression

用tensorflow实现了逻辑回归算法,将sigmoid换成softmax,实现多文本分类的任务。

Primary LanguagePython

tensorflow-logistics-regression

用tensorflow实现的基于逻辑回归的文本分类,将sigmoid函数改为sofamax函数,实现多分类的任务。

环境

python 3.6.2
sklearn
jieba 0.39
tensorflow 1.12.0

数据

数据集采用gaussic大佬的数据集,https://github.com/gaussic/text-classification-cnn-rnn
链接: https://pan.baidu.com/s/1hugrfRu 密码: qfud

运行

训练:python main.py
单条语句测试:python predict.py