Skip to content

Commit ccb8941

Browse files
committed
-- added custom sampler option in fit
-- added utility functions for imbalanced classification -- added future roadmap to github readme
1 parent 130ac54 commit ccb8941

File tree

9 files changed

+520
-189
lines changed

9 files changed

+520
-189
lines changed

README.md

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,16 @@ loaded_model = TabularModel.load_from_checkpoint("examples/basic")
110110

111111
[PyTorch Tabular – A Framework for Deep Learning for Tabular Data](https://deep-and-shallow.com/2021/01/27/pytorch-tabular-a-framework-for-deep-learning-for-tabular-data/)
112112

113+
## Future Roadmap(Contributions are Welcome)
114+
115+
1. Add GaussRank as Feature Transformation
116+
2. Add ability to use custom activations in CategoryEmbeddingModel
117+
3. Add differential dropouts(layer-wise) in CategoryEmbeddingModel
118+
4. Add Fourier Encoding for cyclic time variables
119+
5. Integrate Optuna Hyperparameter Tuning
120+
6. Add Text and Image Modalities for mixed modal problems
121+
7. Integrate Wide and Deep model
122+
8. Integrate TabTransformer
113123
## References and Citations
114124

115125
[1] Sergei Popov, Stanislav Morozov, Artem Babenko. [*"Neural Oblivious Decision Ensembles for Deep Learning on Tabular Data"*](https://arxiv.org/abs/1909.06312). arXiv:1909.06312 [cs.LG] (2019)

docs/.tmp/exp_version_manager.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
classification: 2
1+
classification: 17

docs/history.md

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,11 @@ History
66

77
- First release on PyPI.
88

9-
0.2.0.dev0 (2021-02-03)
9+
0.2.0 (2021-02-07)
1010
------------------
1111

1212
- Fixed an issue with torch.clip and torch version
1313
- Fixed an issue with `gpus` parameter in TrainerConfig, by setting default value to `None` for CPU
14+
- Added feature to use custom sampler in the training dataloader
15+
- Updated documentation and added a new tutorial for imbalanced classification
1416

0 commit comments

Comments
 (0)