chore: poetry update

dreamquark-ai · Mar 23, 2022 · 40107a8 · 40107a8
1 parent 6d0485f
commit 40107a8
Show file tree

Hide file tree

Showing 5 changed files with 133 additions and 106 deletions.
diff --git a/README.md b/README.md
@@ -192,6 +192,12 @@ A complete example can be found within the notebook `pretraining_example.ipynb`.
 
 /!\ : current implementation is trying to reconstruct the original inputs, but Batch Normalization applies a random transformation that can't be deduced by a single line, making the reconstruction harder. Lowering the `batch_size` might make the pretraining easier.
 
+# Data augmentation on the fly
+
+It is now possible to apply custom data augmentation pipeline during training.
+Templates for ClassificationSMOTE and RegressionSMOTE have been added in `pytorch-tabnet/augmentations.py` and can be used as is.
+
+
 # Easy saving and loading
 
 It's really easy to save and re-load a trained model, this makes TabNet production ready.