A machine learning playground using React.js and d3 for visualizing different algorithms that has a low-level dataset creation tool for generating a custom 2-cluster dataset for training. Yuo can train models ranging from SVMs, KNNs, and Naive Bayes to and MLPs (implamented using Pytorch and Scikit-Learn). The app uses a FastAPI backend with a websocket for visualizing model results in real-time and a SQLite database for storing the model data and metrics.
See a demo here (please note that there is no backend running).
Codebase | Description |
---|---|
app | Playground |
fastAPI | fastAPI Backend |
You can install Heart Face the following method:
- Clone the Repo and cd into it
git clone https://github.com/kendreaditya/ml-playground.git cd ml-playground
- Run the frontend app
npm start
- Run the backend
cd backend uvicorn main:app
Whether you have ideas, translations, design changes, code cleaning, or real heavy code changes, help is always welcome. The more is done the better it gets!