basf / mamba-tabular

Mambular is a Python package that brings the power of Mamba architectures to tabular data, offering a suite of deep learning models for regression, classification, and distributional regression tasks. This includes models like Mambular, FT-Transformer, TabTransformer and tabular ResNets.
https://mambular.readthedocs.io
MIT License
133 stars 7 forks source link

adapt config for normalization layer in rnn #152

Closed AnFreTh closed 2 weeks ago

AnFreTh commented 2 weeks ago

This pull request introduces a change to the initialization process of the TabulaRNN class in the tabularnn.py file. The most important changes involve the reordering of the initialization of the normalization layer and the introduction of the replace function from the dataclasses module.

Initialization process improvements: