我已经用test_score
填充了results
,但是我如何将train_score
添加到results
,以便它是一个包含10个元组的列表,例如:(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
你可以这样做:
...
results.append((train_score, test_score))
...
将元组添加到列表中:results.append((train_score,test_score))
另外,请注意for i in list(range(1, 11))
在for循环中是不必要的。for i in range(1, 11)
是足够的,因为它返回一个可迭代的范围对象。
在
之前创建一个元组实例t = (test_score, train_score)
results.append(t)