plots.py

import numpy as np
import matplotlib.pyplot as plt


def plot_decision_boundary(model, X, y, cmap=None, alpha=0.6):
    x_min, x_max = X[:, 0].min() - 1, X[:, 0].max() + 1
    y_min, y_max = X[:, 1].min() - 1, X[:, 1].max() + 1

    xx, yy = np.meshgrid(np.linspace(x_min, x_max, 200),
                         np.linspace(y_min, y_max, 200))

    Z = model.predict(np.c_[xx.ravel(), yy.ravel()])
    Z = Z.reshape(xx.shape)

    plt.figure()
    plt.contourf(xx, yy, Z, cmap=cmap, alpha=alpha)

    plt.xlim(xx.min(), xx.max())
    plt.ylim(yy.min(), yy.max())