Skip to content

Commit 6ba0a80

Browse files
committed
fix: preprocessor argument extended
1 parent 5aa8c12 commit 6ba0a80

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

mambular/models/sklearn_base_regressor.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,11 @@ def __init__(self, model, config, **kwargs):
2525
"task",
2626
"cat_cutoff",
2727
"treat_all_integers_as_numerical",
28-
"knots",
2928
"degree",
29+
"n_knots",
30+
"use_decision_tree_knots",
31+
"knots_strategy",
32+
"spline_implementation",
3033
]
3134

3235
self.config_kwargs = {

0 commit comments

Comments
 (0)