Models are implemented from scratch in PyTorch using only tensor operations.
Models Implemented :
- Logistic Regression (Not technically a deep learning model, but gives a base for Neural Networks)
- Deep Neural Network (Fully-connected layers only)
Note : Only binary classification done till now. Will add multi-class classification also.
Optimization Algorithms Implemented :
- Batch Gradient Descent
- Mini-batch Gradient Descent
- Stochastic Gradient Descent (put
batch_size
= 1 in mini-batch to get this) - Batch Gradient Descent with Momentum
- Batch RMSProp
- Batch Adam
Training and Validation results are added in the Results folder.