如何用KNN模型生成多个预测输出?

2024-06-18 18:50:35 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在使用predict()函数从KNN模型生成预测,而不是有一个预测,在本例中是[10],我希望有最可能的类。你知道吗

有可能吗?你知道吗

这是我的密码:

import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier
from sklearn.metrics import accuracy_score


df = pd.read_excel("Test.xlsx")
X = df.iloc[:,:4]
y = np.array(df['Target']) 

# split into train and test
X_train, 
X_test, 
y_train, 
y_test = train_test_split(X, y, test_size=0.33, random_state=42)

# instantiate learning model (k = 7)
knn = KNeighborsClassifier(n_neighbors=7)

# fitting the model
knn.fit(X_train, y_train)

# predict the response
pred = knn.predict(X_test)

#Predict Output
pred= knn.predict([[2,3,90,600]]) 


**Output:**
**[10]**

Tags: fromtestimportdfmodelasnpneighbors