Skip to content

Commit 2e8d113

Browse files
committed
Change svm_rbf (kernlab) to formula interface, add indicator encodings
1 parent 115292d commit 2e8d113

File tree

1 file changed

+18
-4
lines changed

1 file changed

+18
-4
lines changed

R/svm_rbf_data.R

Lines changed: 18 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,25 +41,39 @@ set_fit(
4141
eng = "kernlab",
4242
mode = "regression",
4343
value = list(
44-
interface = "matrix",
45-
protect = c("x", "y"),
44+
interface = "formula",
45+
protect = c("formula", "data"),
4646
func = c(pkg = "kernlab", fun = "ksvm"),
4747
defaults = list(kernel = "rbfdot")
4848
)
4949
)
5050

51+
set_encoding(
52+
model = "svm_rbf",
53+
eng = "kernlab",
54+
mode = "regression",
55+
options = list(predictor_indicators = FALSE)
56+
)
57+
5158
set_fit(
5259
model = "svm_rbf",
5360
eng = "kernlab",
5461
mode = "classification",
5562
value = list(
56-
interface = "matrix",
57-
protect = c("x", "y"),
63+
interface = "formula",
64+
protect = c("formula", "data"),
5865
func = c(pkg = "kernlab", fun = "ksvm"),
5966
defaults = list(kernel = "rbfdot")
6067
)
6168
)
6269

70+
set_encoding(
71+
model = "svm_rbf",
72+
eng = "kernlab",
73+
mode = "classification",
74+
options = list(predictor_indicators = FALSE)
75+
)
76+
6377
set_pred(
6478
model = "svm_rbf",
6579
eng = "kernlab",

0 commit comments

Comments
 (0)