相关疑难解决方法(0)

是否可以在scikit-learn中使用网格搜索调整参数来定制内核?

我有一个自定义内核函数,我使用GridSearchCV函数和SVC(kernel = my_kernel).

my_kernel函数需要一个参数k来调整,所以我想知道是否可以配置param_grid选项来调整我的自定义内核函数的参数.

例如,可以如下调整RBF内核的gamma参数.我可以为自定义内核提供param_grid = dict(k = k_range)类型的选项吗?

gamma_range = 10. ** np.arange(-5, 4)
param_grid = dict(gamma=gamma_range)
grid = GridSearchCV(SVC(), param_grid=param_grid, cv=StratifiedKFold(y=Y, k=5))
Run Code Online (Sandbox Code Playgroud)

python scikit-learn

7
推荐指数
1
解决办法
3897
查看次数

标签 统计

python ×1

scikit-learn ×1