yorick76ee / tensorflow-logistics-regression

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

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


Languages

Language:Python 92.8%Language:PowerShell 3.7%Language:Batchfile 3.4%