Skip to content

Instantly share code, notes, and snippets.

Created November 24, 2021 07:40
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
What would you like to do?
import pandas as pd
from sklearn.tree import DecisionTreeClassifier
from sklearn.preprocessing import LabelEncoder
from sklearn.model_selection import KFold, cross_val_score
df_drug = pd.read_csv("drug200.csv")
label_encoder = LabelEncoder()
categorical_features = [feature for feature in df_drug.columns if df_drug[feature].dtypes == 'O']
for feature in categorical_features:
X = df_drug.drop("Drug", axis=1)
y = df_drug["Drug"]
model = DecisionTreeClassifier(criterion="entropy"), y)
kfold = KFold(random_state=42, shuffle=True)
cv_results = cross_val_score(model, X, y, cv=kfold, scoring="accuracy")
print(cv_results.mean(), cv_results.std())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment