This visualization uses TensorFlow.js to train a neural network on the titanic dataset and visualize how the predictions of the neural network evolve after every training epoch. The colors of each row indicate the predicted survival probability for each passenger. Red indicates a prediction that a passenger died. Green indicates a prediction that a passenger survived. The intensity of the color indicates the magnitude of the prediction probability. As an example, a bright green passenger represents a strong predicted probability for survival. We also plot the loss of our objective function on the left of the table with D3.js.
- Install Node.js.
- Clone or download this repo.
- Open a terminal and cd into this repo.
- In your terminal type:
- Launch the dev server by typing:
npm run dev
- Click on the url shown in the terminal or enter
http://localhost:8080/in your web browser.
If you wish to bundle the code for production run:
npm run build. This will produce the flat files you can serve on the web in the
/dist/ directory. I have not optimized this code for real production use and the dependencies are quite extraneously large.