https://github.com/mathyouf/ml-loss-landscape-data-visualization-in-webvr
Loss Landscape or a Neural Network Visualized in 3D with A-frame
https://github.com/mathyouf/ml-loss-landscape-data-visualization-in-webvr
Last synced: 2 months ago
JSON representation
Loss Landscape or a Neural Network Visualized in 3D with A-frame
- Host: GitHub
- URL: https://github.com/mathyouf/ml-loss-landscape-data-visualization-in-webvr
- Owner: mathyouf
- License: other
- Created: 2020-08-20T06:12:57.000Z (almost 6 years ago)
- Default Branch: master
- Last Pushed: 2020-08-20T06:13:48.000Z (almost 6 years ago)
- Last Synced: 2026-03-29T14:46:21.616Z (3 months ago)
- Language: JavaScript
- Size: 68.4 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# A-frame with Tensorflow.js
A-frame implementation of 3D plane fitting using Tensorflow.js.
Inspired by the 2D [hello tensorflow](https://glitch.com/edit/#!/hello-tensorflow) project by [Monica Dinculescu](https://glitch.com/@notwaldorf)💖
## Features
- Adjust each x and z values in plane constructed by `x + x^2 + z + z^2`
- Set the number of points in each plane (more improves loss accuracy)
- Set the learning rate
- Choose among optimizers available in [Tensorflow.js](https://js.tensorflow.org/api/latest/#Training-Optimizers)
- Set number of epochs to train for
- Training automatically stops when the loss (MSE) reaches 0.001
## Potential Features
- Vizualize loss with giant Matrix style rectangle that shrinks with the loss
- Code written for this (index.html Line 61) but design is bothersome
- Turn visualization of neural network output into a plane instead of points
- No native method in A-frame, would have to [borow from Three.js](https://stackoverflow.com/questions/51396623/draw-a-curve-using-multi-plane-in-a-frame-and-three-js)
- Add VR UI to adjust these parameters from a VR controller
- Don't own a VR headset currently so I can't test it