Skip to content

feat: scripts to train ml models and compare performance - #32

Merged
selvarajuv merged 3 commits into
mainfrom
26-model-creation-testing
Oct 5, 2026
Merged

selvarajuv merged 3 commits into
mainfrom
26-model-creation-testing

Conversation

@narang-an

@narang-an narang-an commented Sep 29, 2026 •

Copy link
Copy Markdown
Collaborator

Summary

Closes #26
Scripts to train and evaluate 4 different ml models (bayesian, neural network, time series, gradient boot)

Type of change

  • feat — new feature
  • fix — bug fix
  • docs — documentation only
  • refactor — code change that neither fixes a bug nor adds a feature
  • perf — performance improvement
  • test — adding or updating tests
  • chore / build / ci — tooling, deps, or pipeline changes

Changes

  • Implemented basic methods to train and test each model
  • Comparison script to evaluate model performance
  • Generated mock data for tests

Database migrations

  • No schema changes
  • Includes an Alembic migration (just migrate-create "...")
  • Migration runs cleanly (just migrate) and rolls back (just migrate-down)

Checklist

  • Commits follow Conventional Commits
  • Lint passes (just lint) and code is formatted (just format)
  • Pre-commit hooks run clean (just hooks installed)
  • Updated docs / README / .env.example where relevant
  • Tested locally (just dev) and verified the API at http://localhost:8000/docs

How to test

to run tests

  1. cd backend
  2. uv run pytest

to run eval script with fake data

  1. cd backend
  2. python scripts/compare_models.py

Notes for reviewers

@narang-an narang-an linked an issue Sep 29, 2026 that may be closed by this pull request
@narang-an
narang-an force-pushed the 26-model-creation-testing branch from 70014d7 to 728538b Compare September 29, 2026 22:19
@narang-an narang-an changed the title #26 scripts to train ml models and compare performance feat: scripts to train ml models and compare performance Sep 29, 2026
@selvarajuv
selvarajuv force-pushed the 26-model-creation-testing branch from d4a203d to 40e11e5 Compare October 5, 2026 16:21
@selvarajuv
selvarajuv merged commit 718b7ae into main Oct 5, 2026
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Model Creation & Testing

2 participants