我在train_score中发现了一个错误，有人能解决吗？

Question

我在Train_score中有一个错误，但在test_score中没有错误，我如何解决这个问题：

MLA_compare.loc[row_index, 'MLA Time'] = cv_results['fit_time'].mean()
MLA_compare.loc[row_index, 'MLA Train Accuracy Mean'] = cv_results['train_score'].mean()
MLA_compare.loc[row_index, 'MLA Test Accuracy Mean'] = cv_results['test_score'].mean()   

KeyError：'train_score'

Answer 1

必须将条件添加到行中

cv_results = model_selection.cross_validate( alg, data1[data1_x_bin],
                     data1[Target], cv=cv_split, return_train_score=True )