Skip to content

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

Notifications You must be signed in to change notification settings

duguiming111/tensorflow-logistics-regression

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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,实现多文本分类的任务。

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published