我已经填充results
了test_score
,但是如何添加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))
另外,请注意for i in list(range(1, 11))
for 循环中没有必要。for i in range(1, 11)
就足够了,因为它返回一个可迭代的范围对象。
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句