Python - как визуализировать границы решения для LogisticRegression? - PullRequest
0 голосов
/ 10 января 2019

Я пытаюсь визуализировать границу решения для классификатора LogisticRegression ().

Но при наборе plot_boundary(LogisticRegression(), X, y, "Log Reg")

получаю ValueError: X has 2 features per sample; expecting 24

Я проверил X.shape, и это (14635, 24) Что может быть не так с моей функцией?

def plot_boundary(clf, X, y, plot_title): 
    xx, yy = np.meshgrid(np.linspace(-3, 3, 50), np.linspace(-3, 3, 50)) 
    clf.fit(X, y) 
    # plot the decision function for each datapoint on the grid 
    Z = clf.predict_proba(np.vstack((xx.ravel(), yy.ravel())).T)[:, 1] 
    Z = Z.reshape(xx.shape) 

    image = plt.imshow(Z, interpolation='nearest', extent=(xx.min(), xx.max(),
                       yy.min(), yy.max()), aspect='auto', origin='lower', 
                       cmap=plt.cm.PuOr_r) 
    contours = plt.contour(xx, yy, Z, levels=[0], linewidths=2, linetypes='--') 
    plt.scatter(X[:, 0], X[:, 1], s=30, c=y, cmap=plt.cm.Paired) 
    plt.xticks(()) 
    plt.yticks(()) 
    plt.xlabel(r'$x_1$') 
    plt.ylabel(r'$x_2$') 
    plt.axis([-3, 3, -3, 3]) 
    plt.colorbar(image) 
    plt.title(plot_title, fontsize=12)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...