classifiers.py 文件源码

def KNN(X, y):
    print("Iniciando treinamento do KNN")
    clf = KNeighborsClassifier(n_jobs=6,leaf_size=15)
    kf = KFold(len(y),n_folds=20)
    clf.fit(X,y)

    X_score = X[:10000]
    y_score = y[:10000]
    score = clf.score(X_score, y_score)
    print("KNN score: ", score)

    return clf