Skip to content

Krutarth-Bhatt/Neural_Net-Iris_dataset_Classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Neural-Net for classification of Iris dataset::

Hi There, This is a simple neural net for classification of iris flower dataset. Iris flower dataset consists of four different features of a flower which can belong to anyone of the 3 classes which are::

  1. setosa.
  2. virginica.
  3. versicolor.

to know more about Iris Dataset refer the link:: https://en.wikipedia.org/wiki/Iris_flower_data_set

The Architecture:: There are 3 layers from which 1st layer is input layer consisting of four neurons. The middle and also the hidden layer consists of 100 neurons and the last and the final output layer has 3 neurons. I have used Stochastic gradient descent, ie trained over 120 examples at a time in a vectorized manner.

the activations from 1st to 2ad layer is relu and in 2ad-3rd layer is softmax.

This code is written in python 3.6

By running this code output will be accuracy of the trained model over testing data ie test_iris.csv and the graph of Cost function vs no of iteration.

Special Note:: This is a Multipurpose Neural-Net there are three functions params_init, train, and test in these functions any size of neural net can be feed with any other data like mnist or other which can be solved with simple neural net.

Releases

No releases published

Packages

No packages published

Languages