plot_decision_boundary () как-то не дает никакого вывода - PullRequest
0 голосов
/ 12 декабря 2018

, поэтому я пытался просмотреть границу принятия решения для моей сети, и по какой-то причине, когда я запускаю ее, она не дает никакого вывода.Я взял функцию из здесь

, она не выдает никакой ошибки, она просто завершает прогон.

    # Fit the model also history to map the model
history = model.fit(X, Y,validation_split=0.30, epochs=10, batch_size=1000, verbose= 1)
# evaluate the model
scores = model.evaluate(X, Y)

def plot_decision_boundary(X, y, model, steps=1000, cmap='Paired'):
    """
    Function to plot the decision boundary and data points of a model.
    Data points are colored based on their actual label.
    """
    cmap = plt.get_cmap(cmap)

    # Define region of interest by data limits
    xmin, xmax = X[:,0].min() - 1, X[:,0].max() + 1
    ymin, ymax = X[:,1].min() - 1, X[:,1].max() + 1
    steps = 1000
    x_span = np.linspace(xmin, xmax, steps)
    y_span = np.linspace(ymin, ymax, steps)
    xx, yy = np.meshgrid(x_span, y_span)

    # Make predictions across region of interest
    labels = model.predict(np.c_[xx.ravel(), yy.ravel()])

    # Plot decision boundary in region of interest
    z = labels.reshape(xx.shape)

    fig, ax = plt.subplots()
    ax.contourf(xx, yy, z, cmap=cmap, alpha=0.5)

    # Get predicted labels on training data and plot
    train_labels = model.predict(X)
    ax.scatter(X[:,0], X[:,1], c=y, cmap=cmap, lw=0)

    return fig, ax


plot_decision_boundary(X, Y, model, cmap='RdBu')

На самом деле я не внес много изменений в функцию.что мне здесь не хватает?

1 Ответ

0 голосов
/ 12 декабря 2018

Ваша функция plot_decision_boundary() создает объект fig и ax, которые возвращаются в конце.В вашем коде нет ничего, что могло бы занять эти объекты, когда они возвращаются.Просто потому, что функция возвращает fig и ax, которые не означают, они автоматически рисуются.

Решение простое, просто вызовите

plt.show()

после вызова функции границы решения.Эта часть часто опускается в примерах кодов.Я полагаю, что это потому, что существует несколько способов создания окна и отображения графика (вы также можете сохранить его непосредственно в файл, в этом случае вам не понадобится оператор show ()).

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...