在python中从sklearn复制GridSearch失败

Question

我试图在sklearn中做类似于GridSearch的事情:我想获得三个模型的列表，其中所有参数都是固定的，除了C对应于每个模型中的1,10和100。我有以下两个函数。

def params_GridSearch(dic_params):
    keys, values = dic_params.keys(), dic_params.values()
    lst_params = []
    for vs in itertools.product(*values):
        lst_params.append( {k:v for k,v in zip(keys,vs)} )
    return lst_params
def models_GridSearch(model, dic_params):
    models = [ model.set_params(**params) for params in params_GridSearch(dic_params) ]
    return models

然后构建一个模型并指定一个参数字典。

from sklearn.svm import SVC
model = SVC()
dic = {'C': [1,10,100]}

并使用我刚刚定义的函数生成模型。

models = models_GridSearch(model, dic)

然而，重复3次的结果是相同的模型(使用最后一个参数，即100)。似乎有一些混叠正在发生。

Answer 1

model在model_GridSearch中的列表推导的每次迭代中都引用相同的对象，因此您只是将C值分配给同一对象3次。您可以做一些不同的事情来解决这个问题:您可以使用copy模块复制对象，或者将类传递给models_GridSearch函数而不是实例，并在每次迭代中实例化一个对象。您还可以用各种方式重构代码来修复问题。这完全取决于你的目标。

复制方法:

import copy
def models_GridSearch(model, dic_params):
    models = [ copy.deepcopy(model).set_params(**params) for params in params_GridSearch(dic_params) ]
    return models

传入类:

def models_GridSearch(Model, dic_params):
    models = [ Model().set_params(**params) for params in params_GridSearch(dic_params) ]
    return models

from sklearn.svm import SVC
Model = SVC
dic = {'C': [1,10,100]}
models = models_GridSearch(Model, dic)
print models