查找属于特定KNN类的所有输入
查找属于特定KNN类的所有输入
提问于 2022-02-02 09:23:08
我已经创建了一个基于KNN的分类算法，使用sklearn。该算法创建了4个名为"1“、"2”、"3“、"4”的类。我想给算法一个输入列表，并预测它们可能属于哪四个类中的哪个类，并只打印出属于"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)算法正确地分类了输出类，但无法获得所需的列表。这里所需的输出是
L= ["Loop","Loop No."]回答 1
Stack Overflow用户
回答已采纳
发布于 2022-02-03 18:46:55
通过从预测输出中转换列表来获得所需的输出。输入列表和预测列表一起创建一个字典。根据所需的类名对每个键的值进行比较，然后将所需的列表从后面取出。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/70953148
复制相关文章
相似问题
腾讯云开发者
