其他分享
首页 > 其他分享> > 机器学习(二)

机器学习(二)

作者:互联网

一、特征工程之文本特征的抽取API——CountVectorizer

(一)、纯英文文本的计数特征抽取方式

这一api主要是起到文本中某些单词出现的次数进行统计,通过统计文本中某些单词出现的次数来判断这一文章的类型。例如 love等词出现过很多次,可以大体猜测出这一文章主要是情感类文章(仅仅举例,不要抬杠,杠就是你对)。
废话不多说,直接上代码:

from sklearn.feature_extraction.text import CountVectorizer


def txt():
    """
    文本特征的抽取
    :return:
    """
    # 需要处理的两段文本信息
    message = [" There are moments in life when you miss someone so much that you just want to pick them from your dreams and hug them for real!",
               "it was our custom to leave the front door on the latch at night"]

    # 实例化
    CV = CountVectorizer()

    # 调用CountVectorizer()
    data = CV.fit_transform(message)

    print(CV.get_feature_names())

    # 将sparce矩阵转化成数字矩阵并且打印
    print(data.toarray())

以上代码运行的结果为:

['and', 'are', 'at', 'custom', 'door', 'dreams', 'for', 'from', 'front', 'hug', 'in', 'it', 'just', 'latch', 'leave', 'life', 'miss', 'moments', 'much', 'night', 'on', 'our', 'pick', 'real', 'so', 'someone', 'that', 'the', 'them', 'there', 'to', 'want', 'was', 'when', 'you', 'your']
[[1 1 0 0 0 1 1 1 0 1 1 0 1 0 0 1 1 1 1 0 0 0 1 1 1 1 1 0 2 1 1 1 0 1 2 1]
 [0 0 1 1 1 0 0 0 1 0 0 1 0 1 1 0 0 0 0 1 1 1 0 0 0 0 0 2 0 0 1 0 1 0 0 0]]

对以上代码运行的结果做出解释,首先是get_feature_names()这一API的输出:
‘and’, ‘are’, ‘at’, ‘custom’, ‘door’, ‘dreams’, ‘for’, ‘from’, ‘front’, ‘hug’, ‘in’, ‘it’, ‘just’, ‘latch’, ‘leave’, ‘life’, ‘miss’, ‘moments’, ‘much’, ‘night’, ‘on’, ‘our’, ‘pick’, ‘real’, ‘so’, ‘someone’, ‘that’, ‘the’, ‘them’, ‘there’, ‘to’, ‘want’, ‘was’, ‘when’, ‘you’, ‘your’
从这一输出可以看出,sklearn.feature_extraction.text.CountVectorizer()这一API主要是统计文章中所有出现的单词,只要某一单词在此文章中出现过,则在对应的数组位置上标注出出现的次数,否则为零。例如‘and’这一词在第一段中出现过一次,则在第一个数组中的第一个数字置1。相反’and’没有在第二段话中出现过,则在对应的位置置0。注意,文章中出现的单个字母不做统计,例如‘I’ love you 中的‘i’。

(二)、含有中文文本的特征抽取

从纯英文文本的特征抽取方式我们可以看出,CountVectorizer这一接口主要是通过空格来进行拆词的,对于英文来说这并不难,但是对于汉语来说,往往词语之间是没有空格的,这时候我们需要借助一种高效的拆词库——jieba,jiaba可以将一句话中的词语拆分出来,进而达到特征抽取的目的,实现代码如下:

import jieba
from sklearn.feature_extraction.text import CountVectorizer

def Chinese():
    """
    中文文本的特征提取
    :return:
    """
    # 将中文文本通过空格的形式分成词语
    T1 = jieba.cut("一叶知秋,是一个孤单的词。")
    T2 = jieba.cut("铅灰的天空,覆雪的原野,冰封的河面,构成了冷色。")

    # 将词语拼接成列表的形式
    word_list1 = list(T1)
    word_list2 = list(T2)

    # 将列表转换成字符串的形式
    string1 = " ".join(word_list1)
    string2 = " ".join(word_list2)
    
    # 实例化
    CV = CountVectorizer()

    # 将中文文本进行特征提取
    data = CV.fit_transform([string1, string2])

    print(CV.get_feature_names())

    # 将sparce矩阵转化成数字矩阵并且打印
    print(data.toarray())

以上代码运行的结果如下:

['一个', '一叶知秋', '冰封', '冷色', '原野', '天空', '孤单', '构成', '河面', '覆雪']
[[1 1 0 0 0 0 1 0 0 0]
 [0 0 1 1 1 1 0 1 1 1]]

结果的含义以及作用与纯英文文本的特征抽取是一样的了。这段代码可以实现含有汉语词语的文本特征抽取。

标签:CountVectorizer,机器,特征,feature,学习,抽取,文本,CV
来源: https://blog.csdn.net/qq_52302919/article/details/119489251