查找属于特定KNN类的所有输入 [英] Find all inputs belonging to a particular KNN class

问题描述

我已经使用skLearning Python创建了一个基于KNN的分类算法。 该算法已经创建了4个类，命名为&Quot；1&Quot；、&Quot；2&Quot；、&Quot；3&Quot；、&Quot；4&Quot；。 我想给算法提供一个输入列表，并预测它们可能属于四个类中的哪一个类，并仅打印出属于类&1&q；

的那些输入的列表

正在尝试使用：

review_3 = ["Loop","Loop No.", "Customer Tag"]

review_3 = vectorizer.transform(review_3)
print(type(review_3))
L = []

for i in review_3:
     if (knn.predict(i)==1):
         L.append(i)
print(L)

算法正确地对输出类进行了分类，但无法获得所需的列表。 此处所需输出为

L= ["Loop","Loop No."]

推荐答案

通过转换预测输出列表获得了所需的输出。 输入列表和预测列表一起创建词典。根据所需的类名比较每个键的值，然后返回所需的列表。

这篇关于查找属于特定KNN类的所有输入的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！