作家
登录

CNN与RNN对中文文本进行分类--基于TENSORFLOW实现

作者: 来源: 2017-11-23 15:56:57 阅读 我要评论

  •          游戏       0.97      0.96      0.97      1000 
  •          娱乐       0.95      0.98      0.97      1000 
  •  
  • avg / total       0.96      0.96      0.96     10000 
  •  
  • Confusion Matrix... 
  • [[991   0   0   0   2   1   0   4   1   1] 
  •  [  0 992   0   0   2   1   0   5   0   0] 
  •  [  0   1 996   0   1   1   0   0   0   1] 
  •  [  0  14   0 912   7  15   9  29   3  11] 
  •  [  2   9   0  12 892  22  18  21  10  14] 
  •  [  0   0   0  10   1 968   4   3  12   2] 
  •  [  1   0   0   9   4   4 971   0   2   9] 
  •  [  1  16   0   4  18  12   1 941   1   6] 
  •  [  2   4   1   5   4   5  10   1 962   6] 
  •  [  1   0   1   6   4   3   5   0   1 979]] 
  • Time usage: 0:00:05 
  • 在测试集上的精确率达到了96.04%,且各类的precision, recall和f1-score都跨越了0.9。

    大年夜混淆矩阵也可以看出分类效不雅异常优良。

    RNN轮回神经收集

    设备项

    RNN可设备的参数如下所示,在rnn_model.py中。

    1. class TRNNConfig(object): 
    2.     """RNN设备参数""" 
    3.  
    4.     # 模型参数 
    5.     embedding_dim = 64      # 词向量维度 
    6.     seq_length = 600        # 序列长度 
    7.     num_classes = 10        # 类别数 
    8.     vocab_size = 5000       # 词汇表达小 
    9.  
    10.     num_layers= 2           # 隐蔽层层数 
    11.     hidden_dim = 128        # 隐蔽层神经元 
    12.     rnn = 'gru'             # lstm 或 gru 
    13.  
    14.     dropout_keep_prob = 0.8 # dropout保存比例 

        推荐阅读

        教你简单三招永别烦人的Windows 10自动更新

      Tech Neo技巧沙龙 | 11月25号,九州云/ZStack与您一路商量云时代收集界线治理实践 我们在客岁的夏天,也就是微软推送Windows 10周年更新的时刻写过一篇关于若何禁用Windows 10更新的文┞仿>>>详细阅读


      本文标题:CNN与RNN对中文文本进行分类--基于TENSORFLOW实现

      地址:http://www.17bianji.com/lsqh/39190.html

    关键词: 探索发现

    乐购科技部分新闻及文章转载自互联网,供读者交流和学习,若有涉及作者版权等问题请及时与我们联系,以便更正、删除或按规定办理。感谢所有提供资讯的网站,欢迎各类媒体与乐购科技进行文章共享合作。

    网友点评
    自媒体专栏

    评论

    热度

    精彩导读
    栏目ID=71的表不存在(操作类型=0)