SentencesJudger 是一个基于GRU神经网络的句子判断程序,基本的功能是判断文章中的某一句话是否为一个优美的句子。
- 确认Python运行环境
- 安装pyTorch, jieba与paddlepaddle(jieba的词性标注使用)
python3 -m pip install -U torch jieba paddlepaddle
- 运行
main.py
python3 main.py
数据集 | 准确度 |
---|---|
train | 96.7% |
test | 83.5% |
- 确保模型的加载路径
- 使用UTF-8编码
- 内置模型仅支持中文
在train.py
中选择训练样本、测试样本与输出模型路径,运行train.py
即可。训练样本、测试样本为csv文件,格式可参考sample.csv
。
目前模型因样本数量过少,判断效果并不理想,持续更新中。