Skip to content

Instantly share code, notes, and snippets.

@rdisipio

rdisipio/quickstart-4.py

Created November 5, 2022 14:27
Show Gist options
  • Save rdisipio/adae63862ef904abd2867d5206e2080f to your computer and use it in GitHub Desktop.
Save rdisipio/adae63862ef904abd2867d5206e2080f to your computer and use it in GitHub Desktop.
Download ZIP
Raw
quickstart-4.py
from sklearn.tree import DecisionTreeClassifier
from sklearn.model_selection import train_test_split # Import train_test_split function
from sklearn import metrics
(
X_train,
X_test,
y_train,
y_test
) = train_test_split(X, y, test_size=0.3, random_state=1)
clf = DecisionTreeClassifier(random_state=12345)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment