Python For循环卡住了,没有迭代

2024-10-01 05:02:32 发布

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

下面是我为KNN分类器迭代K值的代码。 当我运行这个代码时,循环在运行时被卡住了。 我不得不多次终止程序并重新运行它。 还是不走运!你知道吗

CrossValAccuracy=[]
F1=[]
for i in range(50):

    KNNModel= KNeighborsClassifier(n_neighbors=i+1)
    KNNModel.fit(train_data_x,train_data_y)
    CrossValAccuracy.append(np.mean(cross_val_score(KNNModel, train_data_x, train_data_y, cv=10, scoring='accuracy')))
    F1.append(np.mean(cross_val_score(KNNModel,train_data_x,train_data_y,cv=10,scoring='f1_micro')))

best_k= np.argmax(F1)+1

print ("Best value of K based on model F1-score is:",best_k)

Tags: 代码datanptrainvalmeancvf1