Skip to content

Instantly share code, notes, and snippets.

@liannewriting

liannewriting/decision_tree_confusion_matrix.py

Created July 30, 2020 14:43
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 liannewriting/efaa1be611650bd7563331a24df03bf1 to your computer and use it in GitHub Desktop.
Save liannewriting/efaa1be611650bd7563331a24df03bf1 to your computer and use it in GitHub Desktop.
Download ZIP
random forest machine learning
Raw
decision_tree_confusion_matrix.py
dt = DecisionTreeClassifier(random_state=random_seed)
dt.fit(X_train, y_train)
y_test_pred_dt = dt.predict_proba(X_test)
plot_confusion_matrix(dt, X_test, y_test)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment