sgd-plt.py

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
#%matplotlib widget # Uncomment if working with widgets on a jupyter nb

def plot_sgd(params):
    fig = plt.figure()
    ax = fig.add_subplot(111, projection='3d')
    ax.plot(params[:, 0], params[:, 1], params[:, 2])
    ax.set_xlabel('Bias')
    ax.set_ylabel('Weight 1')
    ax.set_zlabel('Weight 2')
    plt.show()

params = np.array(perceptron.parameters_as_they_change)
plot_sgd(params)