ML Visualizations

01

Neural Network Playground

Interactive visualization of a neural network learning to classify data

Classification 02

Handwriting Recognition

Draw digits and watch MNIST-trained model predict them in real-time

MNIST 03

Image Classification

Classify webcam images using MobileNet trained on ImageNet

MobileNet 04

Pose Detection

Real-time body pose estimation with skeleton overlay

MoveNet 05

Face Mesh 3D

468 facial landmark points tracked in real-time

MediaPipe 06

Style Transfer

Apply artistic styles to images using neural networks

Magenta 07

Hand Tracking

Track 21 hand landmarks for gesture recognition

HandPose 08

Object Detection

Detect and label objects in real-time using COCO-SSD

COCO-SSD 09

Sentiment Visualization

Analyze text sentiment and toxicity with visual feedback

NLP 10

Audio Commands

Recognize spoken words and visualize predictions

Speech 11

Body Segmentation

Real-time person/background segmentation

BodyPix 12

GAN Latent Space

Explore generative adversarial network outputs

GAN 13

Neural Doodle

Transform sketches into images with neural networks

Pix2Pix 14

Music Generation

AI-generated melodies using Magenta.js

Magenta 15

Reinforcement Learning

Watch an AI learn to play a simple game

RL 16

t-SNE Explorer

Dimensionality reduction visualization with perplexity tuning

t-SNE 17

DBSCAN Clustering

Density-based clustering with epsilon neighborhoods

Clustering 18

Mean Shift Clustering

Mode-seeking algorithm with kernel density estimation

Clustering 19

RANSAC Robust Fitting

Random sample consensus for outlier-robust model fitting

Robust Estimation 20

SVM Margin & Kernels

Maximum margin classification with kernel tricks

Classification 21

AdaBoost with Stumps

Adaptive boosting with weight updates and margin visualization

Ensemble 22

Expectation-Maximization

EM algorithm for Gaussian mixture models with step-by-step visualization

Clustering 23

UMAP Explorer

Uniform Manifold Approximation with k-NN graph and fuzzy simplicial sets

Dimensionality Reduction 24

Optimizer Shootout

Compare SGD, AdaGrad, RMSprop, and Adam on loss landscapes

Optimization 25

Baum-Welch HMM Training

Train Hidden Markov Models with forward-backward algorithm visualization

HMM 26

BCJR Soft-Decoder

Trellis-based MAP decoding with log-likelihood ratios for convolutional codes

Coding Theory 27

ELBO Geometry

Visualize Evidence Lower Bound and KL divergence in variational inference

Variational Inference 28

RealNVP Normalizing Flow

Affine coupling layers transform Gaussian to complex distributions

Generative Models 29

Word2Vec in 2D

Train word embeddings on a toy corpus with cosine neighbors and analogies

NLP 30

Bayesian Optimization

Optimize black-box functions with GP surrogate and acquisition functions

Optimization 31

XGBoost Step-by-Step

Watch gradient boosted trees fit residuals iteratively

Ensemble 32

Autodiff Graph Visualizer

Build computation graphs and animate forward/reverse mode automatic differentiation

Autodiff 33

Neural Tangent Kernel

Compare kernel regression with NTK to wide neural networks trained by gradient descent

NTK 34

Grad-CAM Visualization

Visualize CNN attention with gradient-weighted class activation maps

Explainability 35

Jackknife Resampling

Estimate bias and standard error via leave-one-out resampling

Statistics