A Simple In-browser visualisation of polynomial regression algorithms.

Just an attempt to get my hands dirty with TFJS and p5.js, inspired by coding train tfjs playlist.

Add points to the canvas by dragging the mouse. Tweak hyper parameters in the left panel and enjoy!

To try out visit: visualize

Demo

TODO

• Implement regression using TFJS
• Option panel to tweak hyperparameters
• Visualize loss during training using tfjs-vis (Click on Fly!)
• Improved CSS support
• Proper tensor memory management

Contributing

Open to enhancements & bug-fixes

Note

Extreme values of learning rate results in weights becoming NaN