Supports only Fully-Connected layers and Cross-Entropy loss.
- Python 3.x
- NumPy
python3 train.py <train.csv> <test.csv> <submission.csv>
- Add regularization.
- Batch elements balancing.
-
Notifications
You must be signed in to change notification settings - Fork 0
alexey-sidnev/sgd
About
Stochastic Gradient Descent for Binary Classification task
Resources
Stars
Watchers
Forks
Releases
No releases published