search

CodingTrain / Toy-Neural-Network-JS

Neural Network JavaScript library for Coding Train tutorials
MIT License
425 stars 242 forks source link

Doodle classification visualization #94

Open patwork opened 6 years ago

patwork commented 6 years ago

Hello,

I've created simple visualization for doodle classification example. It shows every doodle with coresponding prediction from NN.

Sample screenshot:

doodle

white background = NN was correct red background = guessed "cat" green background = guessed "rainbow" blue background = guessed "train"

In first table in console each row is number of mistakes for every pair label:classification.

Second table is raw list of wrong guesses in format: (index), doodle number, label, classification, guess "cat", guess "rainbow", guess "train".

In overall, looks like NN is pretty consistent with rainbow doodles, but not so much with cats and trains.

shiffman commented 6 years ago

Oh this is awesome, so cool! I think this would be great to incorporate into the example code (or a second more complex version of the example.) I'll keep this open as a reference for future development and videos!