TensorMap

A web application for visually creating machine learning algorithms via drag-and-drop, with reverse engineering to TensorFlow code.

Features

• Drag-and-drop neural network design using ReactFlow
• Reverse engineer visual models to TensorFlow Python code
• Real-time model training with live progress via WebSocket
• CSV and image dataset upload and preprocessing
• Correlation matrix visualization and target field selection

Prerequisites

• Node.js >= 18
• Python 3.12+
• PostgreSQL
• Docker (optional, for containerized setup)

Getting Started

Quick Start (Docker)

# Backend
cd tensormap-backend
docker build -t tensormap-backend .
docker run -p 4300:4300 --env-file .env tensormap-backend

# Frontend
cd tensormap-frontend
docker build -t tensormap-frontend .
docker run -p 3300:3300 tensormap-frontend

Backend

cd tensormap-backend
cp .env.example .env          # Configure DB credentials
uv sync
uv run uvicorn app.main:socket_app --reload --port 4300

Frontend

cd tensormap-frontend
cp .env.example .env          # Optional: configure API URL
npm install
npm start

Service	Port
Frontend	3300
Backend	4300

Running Tests

# Backend
cd tensormap-backend && uv run pytest

# Frontend
cd tensormap-frontend && npm test

Project Structure

tensormap/
  tensormap-backend/    # Python FastAPI server
  tensormap-frontend/   # React + Vite SPA

Contributing

See CONTRIBUTING.md for guidelines.

License

Apache 2.0

Author

Oshan Mudannayake

For questions or queries about this project, please reach out via email.