Merge pull request #217 from basf/dev_fix

adapt base models

Author: AnFreTh

mambular/models/sklearn_base_classifier.py

@@ -27,6 +27,7 @@ class SklearnBaseClassifier(BaseEstimator):
     def __init__(self, model, config, **kwargs):
         self.preprocessor_arg_names = [
             "n_bins",
+            "feature_preprocessing",
             "numerical_preprocessing",
             "categorical_preprocessing",
             "use_decision_tree_bins",

mambular/models/sklearn_base_lss.py

@@ -41,13 +41,15 @@
     PoissonDistribution,
     Quantile,
     StudentTDistribution,
+    JohnsonSuDistribution
 )
 
 
 class SklearnBaseLSS(BaseEstimator):
     def __init__(self, model, config, **kwargs):
         self.preprocessor_arg_names = [
             "n_bins",
+            "feature_preprocessing",
             "numerical_preprocessing",
             "categorical_preprocessing",
             "use_decision_tree_bins",
@@ -373,6 +375,7 @@ def fit(
             "inversegamma": InverseGammaDistribution,
             "categorical": CategoricalDistribution,
             "quantile": Quantile,
+            "johnsonsu": JohnsonSuDistribution,
         }
 
         if distributional_kwargs is None:

mambular/models/sklearn_base_regressor.py

@@ -25,6 +25,7 @@ class SklearnBaseRegressor(BaseEstimator):
     def __init__(self, model, config, **kwargs):
         self.preprocessor_arg_names = [
             "n_bins",
+            "feature_preprocessing",
             "numerical_preprocessing",
             "categorical_preprocessing",
             "use_decision_tree_bins",