add preprocessor args to base classes

AnFreTh · AnFreTh · commit c3006b8d374b · 2024-06-28T09:17:06.000Z
diff --git a/mambular/models/sklearn_base_classifier.py b/mambular/models/sklearn_base_classifier.py
@@ -21,6 +21,8 @@ def __init__(self, model, config, **kwargs):
             "task",
             "cat_cutoff",
             "treat_all_integers_as_numerical",
+            "knots",
+            "degree",
         ]
 
         self.config_kwargs = {
diff --git a/mambular/models/sklearn_base_lss.py b/mambular/models/sklearn_base_lss.py
@@ -43,6 +43,8 @@ def __init__(self, model, config, **kwargs):
             "task",
             "cat_cutoff",
             "treat_all_integers_as_numerical",
+            "knots",
+            "degree",
         ]
 
         self.config_kwargs = {
diff --git a/mambular/models/sklearn_base_regressor.py b/mambular/models/sklearn_base_regressor.py
@@ -20,6 +20,8 @@ def __init__(self, model, config, **kwargs):
             "task",
             "cat_cutoff",
             "treat_all_integers_as_numerical",
+            "knots",
+            "degree",
         ]
 
         self.config_kwargs = {

Original file line number	Diff line number	Diff line change
`@@ -21,6 +21,8 @@ def __init__(self, model, config, **kwargs):`
`21`	`21`	`"task",`
`22`	`22`	`"cat_cutoff",`
`23`	`23`	`"treat_all_integers_as_numerical",`
	`24`	`+ "knots",`
	`25`	`+ "degree",`
`24`	`26`	`]`
`25`	`27`
`26`	`28`	`self.config_kwargs = {`
Original file line number	Diff line number	Diff line change
`@@ -43,6 +43,8 @@ def __init__(self, model, config, **kwargs):`
`43`	`43`	`"task",`
`44`	`44`	`"cat_cutoff",`
`45`	`45`	`"treat_all_integers_as_numerical",`
	`46`	`+ "knots",`
	`47`	`+ "degree",`
`46`	`48`	`]`
`47`	`49`
`48`	`50`	`self.config_kwargs = {`
Original file line number	Diff line number	Diff line change
`@@ -20,6 +20,8 @@ def __init__(self, model, config, **kwargs):`
`20`	`20`	`"task",`
`21`	`21`	`"cat_cutoff",`
`22`	`22`	`"treat_all_integers_as_numerical",`
	`23`	`+ "knots",`
	`24`	`+ "degree",`
`23`	`25`	`]`
`24`	`26`
`25`	`27`	`self.config_kwargs = {`