Skip to content

Instantly share code, notes, and snippets.

@KhyatiMahendru

KhyatiMahendru/decisiontree_entropy.py

Last active June 18, 2019 12:26
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save KhyatiMahendru/5da8ade779f41fddc429fde1bd4b28e9 to your computer and use it in GitHub Desktop.
Save KhyatiMahendru/5da8ade779f41fddc429fde1bd4b28e9 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
decisiontree_entropy.py
from sklearn.tree import DecisionTreeClassifier
clf_entropy = DecisionTreeClassifier(criterion = 'entropy', random_state = 33)
clf_entropy.fit(X, Y)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment