Skip to content

kehsihba19/Neural-Network-Visualizer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Neural-Network-Visualizer

Visualization of handwritten digit classification

Streamlit App

• Classification of handwritten digit dataset using Keras functional API and Tensorflow.

• API call to fetch saved models report using Flask.

• Visualization of different neural network layers and prediction using Streamlit package.

Installation

Refer to the souce code for all the packages needed. Use the package manager pip to install python packages.

pip3 install package_name

Usage

For local setup, clone the repository and make sure to install all the packages beforehand.

Run the API server in the terminal by entering the following command :

python3 ml_server.py

For visulaization using streamlit, move to terminal and enter the command as follows :

streamlit run app.py

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.

Please make sure to update tests as appropriate.

In the live server, API call is hosted on Heroku. It might be slow at times.

Show some ❤️ and ⭐ the repo in case you liked it :)