Skip to content

Latest commit

 

History

History
15 lines (12 loc) · 981 Bytes

README.md

File metadata and controls

15 lines (12 loc) · 981 Bytes
Raw

Fruit-360-classification

In this project, a fully connected Artificial Neural Network(ANN) is implemented from scratch.

Neural network architecture and details

This ANN was implemented to classify 4 classes of fruits. Feedforward algorithm was implemented in vectorized form using softmax as activation function for each layer. Back propagation was implemented in both iterative and vectorized forms with sum of squared errors (SSE) as cost function. Stochastic Gradient Descent algorithm was used to train the network.

ANN

Additional parts included:

  • Hyperparameter tuning
  • Improving SGD using momentum algorithm in this file
  • Adding more classes of fruits and hyperparameter tuning in More Fruits
  • Using softmax as output layer's activation function in this file

Dataset