I've populated results
with test_score
, but how can I add train_score
to results
so that it's a list of 10 tuples like: (train_score
, test_score
)?
import numpy as np
import pandas as pd
from sklearn import tree
from sklearn.tree import DecisionTreeClassifier
results = []
for i in list(range(1, 11)):
clf = DecisionTreeClassifier(max_depth = i, random_state=0).fit(X_train, y_train)
test_score = clf.score(X_test, y_test)
train_score = clf.score(X_train, y_train)
results.append(test_score)
results
CodePudding user response:
You can do something like:
...
results.append((train_score, test_score))
...
CodePudding user response:
Just create a tuple instance before
t = (test_score, train_score)
results.append(t)
CodePudding user response:
Append the tuple to your list :
results.append((train_score,test_score))
Also, note than for i in list(range(1, 11))
is unnecessary in a for loop. for i in range(1, 11)
is sufficient since it returns a range object which is iterable.