我正在使用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]**
您可以添加如下内容:
print(knn.predict_proba(X_test)
这将打印出如下内容:
[x1. x2. x3. x4.]
,显示每个类的概率或置信度。此方法将打印测试集中每个项目的格式。你知道吗相关问题 更多 >
编程相关推荐