【竞赛6】kNN轻松达到0.99以上

按照标杆的格式重新写了下

之前上传的结果达到0.99

供大家讨论

# -*- coding: utf-8 -*-
from sklearn.neighbors import KNeighborsClassifier
import pandas as pd
import numpy as np

train = pd.read_csv('train.csv')
test = pd.read_csv('test.csv')

train.drop('id', axis=1, inplace=True)
test.drop('id', axis=1, inplace=True)

train_labels = train.pop('y')

clf = KNeighborsClassifier()
clf.fit(train, train_labels)

submit = pd.read_csv('sample_submit.csv')
submit['y'] = clf.predict(test)
submit.to_csv('knn', index=False)