Skip to content

Conversation

@AnFreTh
Copy link
Collaborator

@AnFreTh AnFreTh commented Mar 13, 2025

This pull request introduces significant new features and updates to the mambular package, including the addition of the ENODE model and associated utilities. The most important changes include the implementation of the ENODE model, updates to configuration files, and version updates.

New Features:

  • mambular/arch_utils/enode_utils.py: Added the ODSTE and DenseBlock classes, which are key components of the ENODE model. ([mambular/arch_utils/enode_utils.pyR1-R305](https://github.com/OpenTabular/DeepTab/pull/238/files#diff-6a06bbd68b5453cfd81ccc01afeb442756cf43ffe282045b385a681bca3cfa7bR1-R305))
  • mambular/base_models/enode.py: Introduced the ENODE class, a Neural Oblivious Decision Ensemble model for tabular data. ([mambular/base_models/enode.pyR1-R113](https://github.com/OpenTabular/DeepTab/pull/238/files#diff-7141a56634117d6d6cbc030dc7504fa61e7feaba5a3c57f33277de3ebc970e70R1-R113))
  • mambular/models/enode.py: Added ENODERegressor, ENODEClassifier, and ENODELSS classes for different types of ENODE-based models. ([mambular/models/enode.pyR1-R68](https://github.com/OpenTabular/DeepTab/pull/238/files#diff-9750feb98df38f28f1fce11a4040590089c08c28586c7a8ebbb8b37754543ba7R1-R68))

Configuration Updates:

  • mambular/configs/enode_config.py: Created a new configuration file for the ENODE model, defining its hyperparameters. ([mambular/configs/enode_config.pyR1-R48](https://github.com/OpenTabular/DeepTab/pull/238/files#diff-30c1b631a70a22994398f172a639f877130becdc4667523768bd33fbc1c9f0d5R1-R48))
  • mambular/configs/__init__.py: Updated to include the new DefaultENODEConfig. ([mambular/configs/__init__.pyR16-R19](https://github.com/OpenTabular/DeepTab/pull/238/files#diff-2a4058456195c5bb2cd0bb82ba1da912c61110eac0c22bbe4072e2c3f08781edR16-R19))

Version Updates:

  • mambular/__version__.py: Updated the version from 1.2.0 to 1.3.0. ([mambular/__version__.pyL19-R19](https://github.com/OpenTabular/DeepTab/pull/238/files#diff-82475f4d4acfc899c88097b07fa8a0f5c0033dba94a8669a276e143060f14a94L19-R19))
  • pyproject.toml: Updated the package version to 1.3.0. ([pyproject.tomlL3-R3](https://github.com/OpenTabular/DeepTab/pull/238/files#diff-50c86b7ed8ac2cf95bd48334961bf0530cdc77b5a56f852c5c61b89d735fd711L3-R3))

Additional Changes:

  • mambular/base_models/__init__.py: Included the ENODE model in the module exports. ([mambular/base_models/__init__.pyR15-R18](https://github.com/OpenTabular/DeepTab/pull/238/files#diff-1616299d4a3734e6a55c60dbd0477c30faa3b3e1897d5898045ee12a9ac4a856R15-R18))
  • mambular/models/__init__.py: Added ENODEClassifier, ENODELSS, and ENODERegressor to the module exports. ([mambular/models/__init__.pyR30-R35](https://github.com/OpenTabular/DeepTab/pull/238/files#diff-604ef5ccea3a2255e630d11eb803132b1604177f88aa1fd24843d416216070ecR30-R35))

@AnFreTh AnFreTh merged commit d320552 into develop Mar 13, 2025
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants