Skip to content

Add CADRE model with tests, docs, examples, and benchmark#1116

Open
chadwickdonald wants to merge 2 commits intosunlabuiuc:masterfrom
chadwickdonald:master
Open

Add CADRE model with tests, docs, examples, and benchmark#1116
chadwickdonald wants to merge 2 commits intosunlabuiuc:masterfrom
chadwickdonald:master

Conversation

@chadwickdonald
Copy link
Copy Markdown

@chadwickdonald chadwickdonald commented Apr 23, 2026

Contributor

Chadwick Bidwell
bidwell3@illinois.edu

Contribution Type

Model

Original Paper

Predicting Drug Sensitivity of Cancer Cell Lines via
Collaborative Filtering with Contextual Attention
https://proceedings.mlr.press/v126/tao20a/tao20a.pdf

Summary

This PR adds a PyHealth-native implementation of CADRE for multilabel drug-response prediction.

Features include:

  • gene embedding encoder
  • optional contextual attention
  • drug-specific decoder
  • multilabel outputs

Files to Review

  • pyhealth/models/cadre.py
  • pyhealth/models/init.py
  • tests/core/test_cadre.py
  • examples/sample_multilabel_cadre.py
  • examples/benchmark_cadre_vs_mlp.py
  • docs/api/models/pyhealth.models.cadre.rst
  • docs/api/models.rst

Validation

  • unit tests pass
  • example scripts run successfully
  • benchmark compares CADRE variants vs MLP baseline

Notes

This implementation is inspired by the original CADRE paper/repository

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.

1 participant