-
Notifications
You must be signed in to change notification settings - Fork 23
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
845028b
commit 55283d2
Showing
1 changed file
with
28 additions
and
8 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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. |