Question

我正在尝试在sklearn中执行与GridSearch类似的操作：我想获得三个模型的列表，其中除了对应于每个模型中的1,10和100的C之外，所有参数都是固定的。我有以下两个功能。

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)

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

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

无法从python中的sklearn重现GridSearch

1 个答案: