查找属于特定 KNN class 的所有输入

Question

我使用 sklearn Python 创建了一个基于 class 化算法的 KNN。该算法创建了 4 class 个名为“1”、“2”、“3”、“4”的元素。我想为算法提供一个输入列表，并预测它们可能属于四个 class 中的 class，并打印出仅属于 class 的那些输入的列表” 1

正在尝试使用：

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)

算法正确 class 输出 classes 但无法获得所需的列表。这里需要的输出是

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

Answer 1

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

查找属于特定 KNN class 的所有输入

Find all inputs belonging to a particular KNN class

python

list

machine-learning

output

knn