Что такое clf и легенда в plot_decision_regions? - PullRequest
0 голосов
/ 26 мая 2020
from mlxtend.plotting import plot_decision_regions

def knn_comparision(data, k):
    X = data[['x1','x2']].values
    y = data['y'].astype(int).values
    clf = neighbors.KNeighborsClassifier(n_neighbors=k)
    clf.fit(X, y)

    # Plotting decision regions
    plot_decision_regions(X, y, clf=clf, legend=2)

Какие параметры ' clf ' и ' legend ' в plot_decision_regions?

1 Ответ

1 голос
/ 26 мая 2020

clf - это объект классификатора, возвращаемый из neighbors.KNeighborsClassifier, который, вероятно, исходит из sklearn .

BigBen уже связал документацию для plot_decision_regions, объясняющая, что они делают.

...