如何将我的数据用于sklearn.naive_bayes.GaussianNB

Question

我正在尝试理解Naive Bayes是如何工作的，并想使用sklearn.naive_bayes.GaussianNB创建一些示例。我有一个数据：

[
[1, 0],
[1, 1],
[0, 1],
// ...
];

这里[[1, 0]]的第一个数字是宠物的类型:猫或狗，第二个数字:浅颜色或深颜色有宠物。然后我想给出一个像0,0,1这样的列表，它代表了一种颜色，并预测了宠物的类型。

我正在尝试使用http://scikit-learn.org/stable/modules/generated/sklearn.naive_bayes.GaussianNB.html中的示例。我重写了这段代码：

import numpy as np
X = np.array([[-1, -1], [-2, -1], [-3, -2], [1, 1], [2, 1], [3, 2]])
Y = np.array([1, 1, 1, 2, 2, 2])

至：

import numpy as np
X = np.array([ [1, 0], [1, 1], [0, 1] ])

但在我的情况下，应该是什么而不是Y = np.array([1, 1, 1, 2, 2, 2])呢？

或者我应该用另一种格式保存我的数据？