Skip to content

Commit d9b6778

Browse files
committed
Allow additional spelling versions of poly and chi_squared in SVC and SVR.
1 parent 3bc0137 commit d9b6778

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

bindings/Python/sklearn_svc.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,15 +166,15 @@ void parse_provided_kwargs(svc &self, const py::kwargs &args) {
166166
plssvm::kernel_function_type kernel{};
167167
if (kernel_str == "linear") {
168168
kernel = plssvm::kernel_function_type::linear;
169-
} else if (kernel_str == "poly") {
169+
} else if (kernel_str == "poly" || kernel_str == "polynomial") {
170170
kernel = plssvm::kernel_function_type::polynomial;
171171
} else if (kernel_str == "rbf") {
172172
kernel = plssvm::kernel_function_type::rbf;
173173
} else if (kernel_str == "sigmoid") {
174174
kernel = plssvm::kernel_function_type::sigmoid;
175175
} else if (kernel_str == "laplacian") {
176176
kernel = plssvm::kernel_function_type::laplacian;
177-
} else if (kernel_str == "chi_squared") {
177+
} else if (kernel_str == "chi_squared" || kernel_str == "chi-squared") {
178178
kernel = plssvm::kernel_function_type::chi_squared;
179179
} else if (kernel_str == "precomputed") {
180180
throw py::value_error{ R"(The "kernel = 'precomputed'" parameter for the 'SVC' is not implemented yet!)" };

bindings/Python/sklearn_svr.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -122,15 +122,15 @@ void parse_provided_kwargs(svr &self, const py::kwargs &args) {
122122
plssvm::kernel_function_type kernel{};
123123
if (kernel_str == "linear") {
124124
kernel = plssvm::kernel_function_type::linear;
125-
} else if (kernel_str == "poly") {
125+
} else if (kernel_str == "poly" || kernel_str == "polynomial") {
126126
kernel = plssvm::kernel_function_type::polynomial;
127127
} else if (kernel_str == "rbf") {
128128
kernel = plssvm::kernel_function_type::rbf;
129129
} else if (kernel_str == "sigmoid") {
130130
kernel = plssvm::kernel_function_type::sigmoid;
131131
} else if (kernel_str == "laplacian") {
132132
kernel = plssvm::kernel_function_type::laplacian;
133-
} else if (kernel_str == "chi_squared") {
133+
} else if (kernel_str == "chi_squared" || kernel_str == "chi-squared") {
134134
kernel = plssvm::kernel_function_type::chi_squared;
135135
} else if (kernel_str == "precomputed") {
136136
throw py::value_error{ R"(The "kernel = 'precomputed'" parameter for the 'SVR' is not implemented yet!)" };

0 commit comments

Comments
 (0)