Skip to content

Instantly share code, notes, and snippets.

@abhishek-shrm

abhishek-shrm/rfc_vs_dt-5.py

Created May 5, 2020 08:57
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 abhishek-shrm/55ebcd5ed407721c2a3f94eb32be41b4 to your computer and use it in GitHub Desktop.
Save abhishek-shrm/55ebcd5ed407721c2a3f94eb32be41b4 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
rfc_vs_dt-5.py
# Building Decision Tree
from sklearn.tree import DecisionTreeClassifier
dt = DecisionTreeClassifier(criterion = 'entropy', random_state = 42)
dt.fit(X_train, Y_train)
dt_pred_train = dt.predict(X_train)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment