Python 机器学习数据集中的类名存储在哪里？

Question

我正在使用带有 sklearn 的 Python 3.6 上的 iris 数据集来学习机器学习，但我不明白要检索的类名存储在哪里。在 Iris 中，有 3 个类，每个类包含 50 个观测值。您可以使用多个命令来打印类及其相关的数值：

print(iris.target)
print(iris.target_names)

这将导致输出：

[0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
 0 0 0 0 0 0 0 0 0 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 2 2 2 2 2 2
 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2
 2 2]
['setosa' 'versicolor' 'virginica']

因此可以看出，类是 Setosa、Versicolor 和 Virginica。我不明白这些类名的存储位置，或者在模型中如何调用它们。如果对数据或目标使用 shape 命令，则结果为 (150,4) 和 (150,)，这意味着数据中有 150 个观测值和 4 行，目标中有 150 行。然而，我只是无法用我的想法来弥合这个差距。

我不明白的是类名应该存储在哪里。如果我为口袋妖怪类型制作了一个全新的数据集，并且有冰、火、水、飞行，我可以在哪里存储这些类型？是否也要求它们是数字的，例如 iris，带有 0、1、2、3？

Answer 1

Sklearn 使用自定义类型的对象来存储其数据集，以便它们可以将元数据与原始数据一起存储。

如果你加载虹膜数据集

In [2]: from sklearn import datasets

In [3]: iris = datasets.load_iris()

您可以使用type检查对象的类型：

In [4]: type(iris)
Out[4]: sklearn.utils.Bunch

可以用dir查看对象内部的属性：

In [5]: dir(iris)
Out[5]: ['DESCR', 'data', 'feature_names', 'target', 'target_names']

然后使用. 符号来查看属性本身：

In [6]: type(iris.data)
Out[6]: numpy.ndarray

In [7]: type(iris.target)
Out[7]: numpy.ndarray

In [8]: type(iris.feature_names)
Out[8]: list

如果您想为自己的数据集模拟此结构，则必须定义自己的自定义对象类型来模拟此结构。这将涉及定义您自己的类。