我有一个自定义内核函数,我使用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)