Contextualized Logo


An SKLearn-style toolbox for estimating and analyzing models, distributions, and functions with context-specific parameters

State-of-the-art methods to estimate and explore context-specific models

Comprehensive support for multiple types of models and meta-models, with training and analysis tools

Building blocks to support extending meta-models to contextualize your application-specific model

Why Contextualized?

Easy usage

Access state-of-the-art techniques through an open unified API and rich visualizations.

Flexible and customizable

Understand models using a wide range of explainers and techniques using interactive visuals. Choose your algorithm and easily start experimenting in seconds.

Community-driven open source toolkit

Join the community, request and contribute improvements to make your model the best it can be.

Model Types

Contextualized Regression

Do the features which predict an outcome change from sample to sample?

Contextualized Bayesian Networks

Does the ordering of feature interactions change from sample to sample?

Contextualized Correlation

Do the correlations between features change from sample to sample?

What You Can Do With Contextualized

Find Hidden Heterogeneity

  • Find changes in data distributions with Contextualized Bayesian Networks
  • Find changes in feature correlations with Contextualized Correlation Networks
  • Connect Heterogeneity to Outcomes

  • Identify heterogeneous predictors of outcomes with Contextualized Regression
  • Cluster samples according to reasons for outcome
  • Who Can Benefit from Contextualized?

    Data Scientists

    Understand heterogeneity hiding in your dataset.


    Understand heterogenetiy in the observed populations.

    ML Engineers

    Build models which adapt to new environments and contexts.

    ML Researchers

    Contextualize your novel models.

    Getting Started

    Install Contextualized

    Contribute to Contextualized

    We encourage you to join the effort and contribute feedback, algorithms, ideas and more, so we can improve the toolkit together!