def cv(self, estimator_params):
if self.ptypes != None:
if self.ptypes == 'int':
for key in estimator_params.keys():
estimator_params[key] = int(estimator_params[key])
else:
for key in self.ptypes.keys():
estimator_params[key] = self.ptypes[key](estimator_params[key])
if self.pfixed != None:
for key in self.pfixed.keys():
estimator_params[key] = self.pfixed[key]
if self.plist != None:
for key in self.plist.keys():
estimator_params[key] = self.plist[key][int(estimator_params[key]) - 1]
self.estimator.set_params(**estimator_params)
v = self.estimator.evaluate(self.cv_params['X'])
return v
# self.cv_params['estimator'] = estim
# cvscore = cross_val_score(**self.cv_params)
# return numpy.mean(cvscore)
# --------------------------------------------- // --------------------------------------------- #
评论列表
文章目录