Skip to content
Change the repository type filter

All

    Repositories list

    • scset

      Public
      Jupyter Notebook
      0100Updated May 16, 2025May 16, 2025
    • Jupyter Notebook
      0000Updated Jan 2, 2025Jan 2, 2025
    • Jupyter Notebook
      32100Updated Nov 5, 2024Nov 5, 2024
    • Code for Variation in First-line Type 2 Diabetes Treatment due to eGFR and Provider Preferences: A Novel Statistical Analysis. Christina X Ji, Saul Blecker, Michael Oberst, Ming-Chieh Shih, Leora I Horwitz, and David Sontag.
      Python
      0000Updated Sep 14, 2024Sep 14, 2024
    • Code for evaluating single cell foundation models scBERT and scGPT
      Jupyter Notebook
      54620Updated Aug 25, 2024Aug 25, 2024
    • Code for Large-Scale Study of Temporal Shift in Health Insurance Claims. Christina X Ji, Ahmed M Alaa, David Sontag. CHIL, 2023. https://arxiv.org/abs/2305.05087
      Python
      0460Updated Aug 14, 2024Aug 14, 2024
    • Code for Seq-to-Final: A Benchmark for Tuning from Sequential Distributions to a Final Time Point. Christina X Ji, Ahmed M Alaa, David Sontag. https://arxiv.org/abs/2407.09642
      Python
      0000Updated Jul 9, 2024Jul 9, 2024
    • Code for "Finding Regions of Heterogeneity in Decision-Making via Expected Conditional Covariance" at NeurIPS 2021 https://arxiv.org/abs/2110.14508
      Jupyter Notebook
      1300Updated Jun 18, 2024Jun 18, 2024
    • analysis code to reproduce results in NMI submission
      Jupyter Notebook
      1200Updated Jun 1, 2024Jun 1, 2024
    • co-llm

      Public
      Co-LLM: Learning to Decode Collaboratively with Multiple Language Models
      Python
      911650Updated May 7, 2024May 7, 2024
    • Onboarding Humans to work with AI: Algorithms to find regions and describe them in natural language that show how humans should collaborate with AI (NeurIPS23)
      Jupyter Notebook
      21200Updated Mar 15, 2024Mar 15, 2024
    • SCOPE

      Public
      Codebase for SCOPE architecture.
      Python
      2200Updated Feb 7, 2024Feb 7, 2024
    • Human-AI Deferral Evaluation Benchmark (Learning to Defer) AISTATS23
      Python
      12110Updated Jan 8, 2024Jan 8, 2024
    • Code and webpages for our study on teaching humans to defer to an AI
      Jupyter Notebook
      01100Updated Nov 6, 2023Nov 6, 2023
    • TabLLM

      Public
      Python
      57318170Updated Nov 2, 2023Nov 2, 2023
    • Python package for machine learning for healthcare using a OMOP common data model
      Python
      2711020Updated Jun 17, 2023Jun 17, 2023
    • Jupyter Notebook
      0000Updated May 17, 2023May 17, 2023
    • Code for paper, "Falsification of Internal and External Validity in Observational Studies via Conditional Moment Restrictions." This repo is currently under construction. Check back later for end-to-end notebooks recreating the results of our AISTATS 2023 paper.
      Jupyter Notebook
      1200Updated Apr 22, 2023Apr 22, 2023
    • vae_ssl

      Public
      Scalable semi-supervised learning with deep variational autoencoders
      Jupyter Notebook
      2902Updated Feb 10, 2023Feb 10, 2023
    • Code for Sample Efficient Learning of Predictors that Complement Humans (ICML 2022)
      Python
      2500Updated Jan 28, 2023Jan 28, 2023
    • Code for "Consistent Estimators for Learning to Defer to an Expert" (ICML 2020)
      Jupyter Notebook
      613015Updated Jan 28, 2023Jan 28, 2023
    • cfrnet

      Public
      Counterfactual Regression
      Python
      8331330Updated Dec 7, 2022Dec 7, 2022
    • ief

      Public
      Train pharmacodynamic deep generative models to model disease progression
      Jupyter Notebook
      1600Updated Dec 2, 2022Dec 2, 2022
    • ml_mmrf

      Public
      Machine Learning with data from the Multiple Myeloma Research Foundation
      Jupyter Notebook
      1820Updated Dec 2, 2022Dec 2, 2022
    • Code for paper, "Falsification before Extrapolation in Causal Effect Estimation"
      Jupyter Notebook
      2300Updated Sep 27, 2022Sep 27, 2022
    • Code for co-training large language models (e.g. T0) with smaller ones (e.g. BERT) to boost few-shot performance
      Python
      31700Updated Sep 23, 2022Sep 23, 2022
    • Code for paper "Evaluating Robustness to Dataset Shift via Parametric Robustness Sets"
      Python
      5800Updated Jul 6, 2022Jul 6, 2022
    • wilds

      Public
      A machine learning benchmark of in-the-wild distribution shifts, with data loaders, evaluators, and default models.
      Python
      131000Updated Apr 4, 2022Apr 4, 2022
    • prancer

      Public
      Platform enabling Rapid Annotation for Clinical Entity Recognition
      JavaScript
      125018Updated Mar 29, 2022Mar 29, 2022
    • Code for ICML 2021 paper "Regularizing towards Causal Invariance: Linear Models with Proxies" (ICML 2021)
      Jupyter Notebook
      21100Updated Mar 14, 2022Mar 14, 2022