Skip to content

Instantly share code, notes, and snippets.

@WillKoehrsen

WillKoehrsen/decision_tree.py

Last active December 29, 2019 19:55
Show Gist options
  • Save WillKoehrsen/eaa3f695de592be52bf4aef6819f3383 to your computer and use it in GitHub Desktop.
Save WillKoehrsen/eaa3f695de592be52bf4aef6819f3383 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
decision_tree.py
from sklearn.tree import DecisionTreeClassifier
# Make a decision tree and train
tree = DecisionTreeClassifier(random_state=RSEED)
tree.fit(X, y)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment