Skip to content

Commit

Permalink
updated README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
Ishan-Kumar2 authored Jul 29, 2020
1 parent 845028b commit 55283d2
Showing 1 changed file with 28 additions and 8 deletions.
36 changes: 28 additions & 8 deletions ANN/README.md
Original file line number Diff line number Diff line change
@@ -1,10 +1,30 @@
# TensorFlow.js Example: Iris Classification using tfjs-data APIs
## Artificial Neural Network
This example uses a fully connected neural network to classify tabular data representing different flowers, running purely in the browser using TensorFlow.js. The goal is to predict what kind of flower it is based on those features of each data point. The data comes from the famous Iris flower data set. The hyperparameters and optimizer used in training the network can be changed and weights learned can be visualised.

## Features
* The number of layers and number of neurons per layer can be changed and the architecture visualised.
effect on the principal components generated.
* Effect of hyperparamters like learning rate and batch size on training can be compared.
* The oprimizer can be choosen from Adam RMSProp and SGD.
* The trained model can be tested on a user defined datal point.
* The weights of network can be visualised along with their magnitude.

## Running locally for development

To run it locally, you must install Python 3 or above and run the following command at the repository's root to set up a python server.

```python
python -m http.server
```

You can then browse to `localhost:8000` and select index.html to view the application.

## Contributors
1. Shashwat Gupta
2. Sahil Goyal
3. Ishan Kumar




This example is identical to the
[iris](https://github.com/tensorflow/tfjs-examples/tree/master/iris) example,
except that it uses the [tfjs-data](https://github.com/tensorflow/tfjs-data)
APIs.

Please see the
[iris](https://github.com/tensorflow/tfjs-examples/tree/master/iris) example for
details and how to run.

0 comments on commit 55283d2

Please sign in to comment.