search

zackhy / TextClassification

Text classification using different neural networks (CNN, LSTM, Bi-LSTM, C-LSTM).
MIT License
194 stars 60 forks source link

你好，请问如何预测单个句子？ #6

Closed albertyou2 closed 6 years ago

albertyou2 commented 6 years ago

@zackhy 作者您好： 很易用的库。我几分钟就用起来了，同时用的是自己的数据，目前准确率也不错！！赞一个 但是您的 test.py是一个测试一组数据并给出准确率的方法。 请问如何写一个最小的预测的python脚本？ 例如 ： python predict.py '需要预测分类的句子' ，输出是 分类标签例如 ： 2 刚刚接触tensorflow不久，望您指导下，谢谢！

zackhy commented 6 years ago

@albertyou2 可以用tf.flags从命令行接收单个句子，把batch_size设为1，将单句按vocabulary转换为vector，然后restore graph，输入vector，得到prediction，把test.py修改一下即可。

albertyou2 commented 6 years ago

@zackhy 谢谢，另外我好像没看到在哪里设置学习率的衰减策略？

zackhy commented 6 years ago

@albertyou2 是的，我没有用learning rate decay。

albertyou2 commented 6 years ago

@zackhy 请问有没有方法可以简单的加入一个 衰减的 方法？ 我发现目前训练到72% 上不去了，怀疑是学习率的原因，谢谢您

zackhy commented 6 years ago

@albertyou2 这个网上教程挺多的，你可以google一下，设一个decay rate，然后每几轮decay一次，改一下train.py就可以了。我过几天有空的话可能会把这个加进去~

albertyou2 commented 6 years ago

好的谢谢您！

zackhy commented 6 years ago

@albertyou2 加了learning rate decay