当前位置:   article > 正文

深度学习|中文文本的分类(处理篇)

token与中文文本转换
3629157-3e755bc73fe9f558.png

前言

之前我们通过朴素贝叶斯方法,做过英文文档的分类(传送门)。那使用中文文本,如何使用深度学习方法来进行分类了?这就是本文所讲的。首先我们来看看中文文本和英文文本的不同。
在处理英文文本时,我们使用的是TF-IDF方法,该方法当然也可以使用在中文文本中,但是我们都知道,中文的分词不像英文那样,每个词都是通过空格分开的,中文我们通过jieba来进行分词。

数据处理

我们数据来源于王老师的一篇文章的数据,是某商品的评价信息。我们处理的大概流程如下:

  • 数据情况
  • 中文文本分词
  • 建立token
  • token转换为列表
  • 统一长度

具体步骤见下。

数据情况

首先我们通过pandas读取我们的数据。数据就是评论文本和评论的情绪分类(0位消极,1位积极)。

  1. import pandas as pd
  2. data = pd.read_csv('dianping.csv')
  3. data.head()
3629157-bd2d784237044d34.png
文本分词

这里我们使用jieba对数据进行分词。

  1. import jieba
  2. data['text'] = data.comment.apply(lambda x: " ".join(jieba.cut(x)))
3629157-2c56df6b4d005262.png

我们保留text和sentiment即可。

建立token

文本数据电脑是没法识别的,电脑只知道数字信息,所以接下来我们就要把文本数据转换为数字,有很多方法,之前的TF-IDF就是其中一种方法。我们这次换一种方法:我们按照分词出现的次数排序建立字典,也就是所谓的token,我们通过keras来实现。

  1. from keras.preprocessing.text import Tokenizer
  2. from keras.preprocessing.sequence import pad_sequences
  3. import numpy as np
  4. tokenizer = Tokenizer(num_words=10000)
  5. tokenizer.fit_on_texts(data.text)
  6. tokenizer.word_index
3629157-800bfc22e407ab34.png
转换为列表

接下来我们将其转换为列表,这样就是数字信息了。

sequences = tokenizer.texts_to_sequences(data.text)
统一长度

我们都知道,评论的长度不一样,数据也就不一样,我们截长补短。

seq = pad_sequences(sequences, maxlen=100)

划分数据集

最后,我们划分数据集,整个的处理过程也就结束了。

  1. X = seq
  2. Y = data['sentiment']
  3. from sklearn.model_selection import train_test_split
  4. X_train, X_test, y_train, y_test = train_test_split(X, Y, test_size=0.2, random_state=33)
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/一键难忘520/article/detail/847258
推荐阅读
相关标签
  

闽ICP备14008679号