python - 将 dict 传递给 scikit learn estimator

Question

我试图将模型参数作为 dict 传递给 Scikit-learn 估计器，但没有运气。它似乎只是将我的 dict 嵌套到参数之一中。例如:

params = {
 'copy_X': True, 
 'fit_intercept': False, 
 'normalize': True
}

lr = LinearRegression(params)

给我:

LinearRegression(copy_X=True,
         fit_intercept={'copy_X': True, 'fit_intercept': False,'normalize': True},
     normalize=False)

另外，我创建了一个函数来迭代字典，并可以创建一个字符串，如:

'copy_X=True, fit_intercept=True, normalize=False'

这同样不成功。有人在这里有什么建议吗？我唯一的限制是数据将以 dict 的形式传给我(实际上是一个 json 对象正在加载 json.uploads)。

谢谢。

Answer 1

使用正确参数初始化估算器的最佳解决方案是解压字典:

lr = LinearRegression(**params)

如果由于某种原因之后需要设置一些参数，可以使用:

lr.set_params(**params)

这比使用 setattr 的优势在于它允许 Scikit 学习对参数执行一些验证检查。