Skip to content

README.md

Latest commit

 

History

History
91 lines (64 loc) · 1.87 KB

README.md

File metadata and controls

91 lines (64 loc) · 1.87 KB

TensorMap

License

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.