Repository navigation
v1.1.0
What's Changed
From Changelog:
- Added ADAM optimizer to go along with SGD: Initial implementation of Adam with tests against Kaggle Cancer dataset.
- Ported over the Kaggle Titanic Project to conduct integration tests and test accuracy improvements for new hyperparam additions in this release.
- Added Regression Capabilities to go along with existing multi-class and binary Classification capabilities. Testing regression feature using Streeteasy NYC dataset.
- He initialization
- Enums for Classification/Regression & Optimizer type
- Pydantic for Config class
- Warmup
- Dropout
- Stubbed the He-init. Performance is not improving for current test. O… by @DeepBlockDeepak in #15
- Mini batch working. by @DeepBlockDeepak in #19
- Regression capabilities added. by @DeepBlockDeepak in #20
- tqdm Loading bar used for training; early stopping incorporated with … by @DeepBlockDeepak in #22
- New Tests with customizing each config with C and warmup_steps by @DeepBlockDeepak in #23
Full Changelog: v1.0.1...v1.1.0