树叶云AI教程：AI人工智能 在Python中构建分类器的步骤

在本章中，我们将重点讨论监督式学习 – 分类。

分类技术或模型试图从观测值中得出一些结论。 在分类问题中，我们有分类输出，如“黑色”或“白色”或“教学”和“非教学”。 在构建分类模型时，需要有包含数据点和相应标签的训练数据集。 例如，如果想检查图像是否属于汽车。 要实现这个检查，我们将建立一个训练数据集，其中包含与“车”和“无车”相关的两个类。 然后需要使用训练样本来训练模型。 分类模型主要用于人脸识别，垃圾邮件识别等。

为了在 Python 中构建分类器，将使用 Python 3 和 Scikit-learn，这是一个用于机器学习的工具。 按照以下步骤在 Python 中构建分类器 –

第1步 – 导入 Scikit-learn 这将是在 Python 中构建分类器的第一步。 在这一步中，将安装一个名为 Scikit-learn 的 Python包，它是 Python中最好的机器学习模块之一。 以下命令导入包 –

import sklearn

第2步 – 导入 Scikit-learn 的数据集

在这一步中，我们可以开始使用机器学习模型的数据集。 在这里，将使用乳腺癌威斯康星诊断数据库。 数据集包括有关乳腺癌肿瘤的各种信息，以及恶性或良性分类标签。 该数据集在 569 个肿瘤上具有 569 个实例或数据，并且包括关于 30 个属性或特征(诸如肿瘤的半径，纹理，光滑度和面积)的信息。 借助以下命令，导入 Scikit-learn的乳腺癌数据集 –

from sklearn.datasets import load_breast_cancer

现在，以下命令将加载数据集。

data = load_breast_cancer()

以下是字典键列表 –

• 分类标签名称(target_names)
• 实际标签(目标)
• 属性/功能名称(feature_names)
• 属性(数据)

现在，使用以下命令，可以为每个重要信息集创建新变量并分配数据。 换句话说，可以用下列命令组织数据 –

label_names = data['target_names']
labels = data['target']
feature_names = data['feature_names']
features = data['data']

现在，为了使它更清晰，可以使用以下命令来打印类标签，第一个数据实例的标签，我们的功能名称和功能的值 –

print(label_names)

上述命令将分别打印恶性和良性的分类名称。输出结果如下 –

['malignant' 'benign']

现在，下面的命令将显示它们被映射到二进制值 0 和 1。这里 0 表示恶性肿瘤，1 表示良性癌症。得到以下输出 –

print(labels[0])
0

下面给出的两个命令将生成功能名称和功能值。

print(feature_names[0])
mean radius
print(features[0])
[ 1.79900000e+01 1.03800000e+01 1.22800000e+02 1.00100000e+03
  1.18400000e-01 2.77600000e-01 3.00100000e-01 1.47100000e-01
  2.41900000e-01 7.87100000e-02 1.09500000e+00 9.05300000e-01
  8.58900000e+00 1.53400000e+02 6.39900000e-03 4.90400000e-02
  5.37300000e-02 1.58700000e-02 3.00300000e-02 6.19300000e-03
  2.53800000e+01 1.73300000e+01 1.84600000e+02 2.01900000e+03
  1.62200000e-01 6.65600000e-01 7.11900000e-01 2.65400000e-01
  4.60100000e-01 1.18900000e-01]

从上面的输出中，可以看到第一个数据实例是一个半径为 1.7990000e + 01 的恶性肿瘤。

第3步 – 组织数据

在这一步中，将把数据分成两部分，即训练集和测试集。 将数据分割成这些集合非常重要，因为必须在未看到的数据上测试模型。要将数据分成集合，sklearn 有一个叫做 train_test_split() 函数的函数。 在以下命令的帮助下，可以分割这些集合中的数据 –

from sklearn.model_selection import train_test_split

上述命令将从 sklearn 中导入 train_test_split 函数，下面的命令将数据分解为训练和测试数据。 在下面给出的例子中，使用 40% 的数据进行测试，其余数据将用于训练模型。

train, test, train_labels, test_labels = train_test_split(features,labels,test_size = 0.40, random_state = 42)

第4步 – 建立模型 在这一步中，我们将建立模型。使用朴素贝叶斯算法来构建模型。 以下命令可用于构建模型 –

from sklearn.naive_bayes import GaussianNB

上述命令将导入 GaussianNB 模块。 现在，以下命令用来初始化模型。

gnb = GaussianNB()

将通过使用 gnb.fit() 将它拟合到数据来训练模型。

model = gnb.fit(train, train_labels)

第5步 – 评估模型及其准确性 在这一步中，我们将通过对测试数据进行预测来评估模型。为了做出预测，我们将使用 predict() 函数。 以下命令做到这一点 –

preds = gnb.predict(test)
print(preds)
## -- 结果如下
[1 0 0 1 1 0 0 0 1 1 1 0 1 0 1 0 1 1 1 0 1 1 0 1 1 1 1 1 1
 0 1 1 1 1 1 1 0 1 0 1 1 0 1 1 1 1 1 1 1 1 0 0 1 1 1 1 1 0 
 0 1 1 0 0 1 1 1 0 0 1 1 0 0 1 0 1 1 1 1 1 1 0 1 1 0 0 0 0 
 0 1 1 1 1 1 1 1 1 0 0 1 0 0 1 0 0 1 1 1 0 1 1 0 1 1 0 0 0 
 1 1 1 0 0 1 1 0 1 0 0 1 1 0 0 0 1 1 1 0 1 1 0 0 1 0 1 1 0 
 1 0 0 1 1 1 1 1 1 1 0 0 1 1 1 1 1 1 1 1 1 1 1 1 0 1 1 1 0 
 1 1 0 1 1 1 1 1 1 0 0 0 1 1 0 1 0 1 1 1 1 0 1 1 0 1 1 1 0 
 1 0 0 1 1 1 1 1 1 1 1 0 1 1 1 1 1 0 1 0 0 1 1 0 1]

上述 0 和 1 系列是肿瘤类别的预测值 – 恶性和良性。

现在，通过比较两个数组即 test_labels和 preds，可以发现模型的准确性。使用 accuracy_score() 函数来确定准确性。 考虑下面的命令 –

from sklearn.metrics import accuracy_score
print(accuracy_score(test_labels,preds))
0.951754385965

结果显示 NaïveBayes 分类器准确率为 95.17%。

通过这种方式，借助上述步骤，我们可以使用 Python 构建分类器。