Mithin Sagar
Skip to content
MediXplain interface
← All projectsMachine Learning · Explainable AI2026

MediXplain

Calibrated disease prediction with retrieved evidence

83.09%
best model accuracy
~190K
patient records
377
symptoms
727
conditions

“In symptom checking, a confidently wrong answer is worse than no answer.”

The problem

Symptom checkers return a label and a percentage that means nothing — the confidence is uncalibrated, and there is no evidence a user could independently verify.

The approach

Seven scikit-learn classifiers are trained and benchmarked in one command and the best is persisted automatically. Probability calibration aligns reported confidence with observed accuracy, predictions are returned as a top-3 differential rather than one hard label, and a FAISS retrieval layer over 384-dimensional sentence-transformer embeddings attaches supporting medical evidence to every prediction.

What makes it
work.

    01

    Logistic Regression, Linear SVM, Bernoulli NB, Decision Tree, Random Forest, AdaBoost and an MLP, 5-fold cross-validated.

    02

    FAISS vector search over a paragraph knowledge base, with a TF-IDF path retained as a lightweight fallback.

    03

    Optional deep-learning branch with LSTM and GRU sequence models over symptom text.

    04

    GitHub Actions pipeline running linting and tests across a Python 3.10–3.12 matrix on every change.

In the interface.

03 screens

Built with.

  • Python
  • scikit-learn
  • FAISS
  • sentence-transformers
  • Flask
  • RAG
Work together

Let’s build somethingmeaningful.

Open to internships and AI/ML roles, research collaborations, or a conversation about something you are trying to make work. I reply to everything.