Skip to content

vincemaina/train-your-own-neural-network

Repository files navigation

Train your own neural network

Try it out here: https://train-your-own-neural-network.vercel.app/

Demo

You should get something like this once you've trained your model:

Recording2024-09-09141957-ezgif com-video-to-gif-converter

You can see the model is able to correctly classify Xs and Os with over 90% confidence each time.

Visualiser

The app features a helpful visualiser which displays the activations and weights in your neural network.

You'll see patterns emerging in the visualiser as you train your model and it picks out patterns in the input data.

Before:

image

After:

image

Activiations

image

Another thing you'll notice in the visualiser, is the neurons change colour as you draw in the input square.

These colours display activations in the neural network.

A fully green neuron, means it is fully activated with a value of 1. Whereas red indicates a value of 0.

As you plan around with the neural network, you'll notice the activations in the input layer correspond to the input data shown in the top right square. And the hidden (middle) layer then extracts patterns e.g. lines/edges in the drawings.

How to train your model using the tool

Recording.2024-09-09.144117.mp4

About

A neural network that runs in browser and visualises learning in real-time!

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published