Skip to content

Commit 4385952

Browse files
committed
tabularRnn regressor hotfix
1 parent b5b3184 commit 4385952

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

mambular/models/tabularnn.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,9 @@ class TabulaRNNRegressor(SklearnBaseRegressor):
8686
The number of knots to be used in spline transformations.
8787
"""
8888

89+
def __init__(self, **kwargs):
90+
super().__init__(model=TabulaRNN, config=DefaultTabulaRNNConfig, **kwargs)
91+
8992

9093
class TabulaRNNClassifier(SklearnBaseClassifier):
9194
"""

0 commit comments

Comments
 (0)