下面是我为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)
目前没有回答
相关问题 更多 >
编程相关推荐