My assumption is that my training set includes the test set, but I don't know how to change this.
from sklearn.model_selection import train_test_split
import sklearn.metrics as metrics
# dataframe to store model performances
scores=pd.DataFrame([],columns=['model', 'recall', 'f1', 'accuracy'])
from sklearn.tree import DecisionTreeClassifier
cv_scores_ac=[]
cv_scores_f1=[]
cv_scores_re=[]
for cv in range(1, 6):
print ("Decision Tree - Iteration %i" % cv)
X_train, X_test, y_train, y_test = train_test_split(X, y, stratify=y, random_state=cv)
tree=DecisionTreeClassifier(min_samples_split=20)
tree.fit(X_train, y_train)
y_pred=tree.predict(X_test)
cv_scores_ac.append(metrics.accuracy_score(y_test, y_pred))
cv_scores_f1.append(metrics.f1_score(y_test, y_pred))
cv_scores_re.append(metrics.recall_score(y_test, y_pred))