Skip to content

Instantly share code, notes, and snippets.

@jovianlin

jovianlin/plot_decision_boundary.py

Created December 22, 2016 14:32
Show Gist options
  • Save jovianlin/5eca057397144781e54ee18f03a6429d to your computer and use it in GitHub Desktop.
Save jovianlin/5eca057397144781e54ee18f03a6429d to your computer and use it in GitHub Desktop.
Download ZIP
Plot Decision Boundary
Raw
plot_decision_boundary.py
# Train the logistic rgeression classifier
clf = sklearn.linear_model.LogisticRegressionCV()
clf.fit(X, y)
# Plot the decision boundary
plot_decision_boundary(lambda x: clf.predict(x))
plt.title("Logistic Regression")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment