在scikit-learn中实现K邻居分类器,每个对象有3个特征

我想使用scikit-learn模块实现一个KNeighborsClassifier(http://scikit-learn.org/dev/modules/generated/sklearn.neighbors.KNeighborsClassifier.html

我从图像中提取了固体度、延展性和胡矩特征。我该如何准备这些数据用于训练和验证?我是否需要为从图像中提取的每个对象创建一个包含3个特征[Hm, e, s]的列表(一张图像中有多个对象)?

我阅读了这个例子(http://scikit-learn.org/dev/modules/generated/sklearn.neighbors.KNeighborsClassifier.html):

X = [[0], [1], [2], [3]]y = [0, 0, 1, 1]from sklearn.neighbors import KNeighborsClassifierneigh = KNeighborsClassifier(n_neighbors=3)neigh.fit(X, y) print(neigh.predict([[1.1]]))print(neigh.predict_proba([[0.9]]))

X和y是两个特征吗?

samples = [[0., 0., 0.], [0., .5, 0.], [1., 1., .5]]from sklearn.neighbors import NearestNeighborsneigh = NearestNeighbors(n_neighbors=1)neigh.fit(samples) print(neigh.kneighbors([1., 1., 1.])) 

为什么第一个例子使用X和y,而现在使用sample?


回答:

你的第一段代码定义了一个在1d数据上的分类器。

X代表特征向量。

[0] 是第一个数据示例的特征向量[1] 是第二个数据示例的特征向量....[[0],[1],[2],[3]] 是所有数据示例的列表,每个示例只有1个特征。

y代表标签。

下图展示了这个概念:

enter image description here

  • 绿色节点是标签为0的数据
  • 红色节点是标签为1的数据
  • 灰色节点是标签未知的数据。
    print(neigh.predict([[1.1]]))

这是在请求分类器为x=1.1预测一个标签。

    print(neigh.predict_proba([[0.9]]))

这是在请求分类器为每个标签提供成员概率估计。

由于两个灰色节点更靠近绿色节点,因此下面的输出是有意义的。

    [0] # 绿色标签    [[ 0.66666667  0.33333333]]  # 绿色标签的概率更大

第二段代码实际上对scikit-learn有很好的说明:

在下面的例子中,我们从表示数据集的数组中构建一个NeighborsClassifier类,并询问谁离[1,1,1]最近

>>> samples = [[0., 0., 0.], [0., .5, 0.], [1., 1., .5]]>>> from sklearn.neighbors import NearestNeighbors>>> neigh = NearestNeighbors(n_neighbors=1)>>> neigh.fit(samples) NearestNeighbors(algorithm='auto', leaf_size=30, ...)>>> print(neigh.kneighbors([1., 1., 1.])) (array([[ 0.5]]), array([[2]]...))

这里没有目标值,因为这只是一个NearestNeighbors类,它不是一个分类器,因此不需要标签。

对于你自己的问题:

由于你需要一个分类器,如果你想使用KNN方法,你应该使用KNeighborsClassifier。你可能需要像下面这样构建你的特征向量X和标签y

X = [ [h1, e1, s1],       [h2, e2, s2],      ...    ]y = [label1, label2, ..., ]

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注