Merge pull request #110 from basf/hotfix

Hotfix
basf · Aug 13, 2024 · af6641b · af6641b
2 parents a327ef8 + 4385952
commit af6641b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/mambular/models/sklearn_base_classifier.py b/mambular/models/sklearn_base_classifier.py
@@ -316,7 +316,7 @@ def fit(
         self : object
             The fitted classifier.
         """
-        if not self.built and not rebuild:
+        if rebuild:
             if not isinstance(X, pd.DataFrame):
                 X = pd.DataFrame(X)
             if isinstance(y, pd.Series):

diff --git a/mambular/models/tabularnn.py b/mambular/models/tabularnn.py
@@ -86,6 +86,9 @@ class TabulaRNNRegressor(SklearnBaseRegressor):
         The number of knots to be used in spline transformations.
     """
 
+    def __init__(self, **kwargs):
+        super().__init__(model=TabulaRNN, config=DefaultTabulaRNNConfig, **kwargs)
+
 
 class TabulaRNNClassifier(SklearnBaseClassifier):
     """