This repository contains code and notebooks for simulation for the paper Multicategory Partial Identification Learning for Individualized Treatment Rules.
helper.py— Utility functions for simulationsmcat_learn.py— Main learning algorithms and methodssimluation.ipynb— Main simulation notebookpixi.toml— Pixi environment configuration
This project uses Pixi for reproducible environment management. To get started:
- Install Pixi: https://pixi.prefix.dev/latest/
- Run
pixi installin the project directory to set up the environment.
- Run the simulation notebooks in your Jupyter environment and select the pixi kernel (which is located in you project directory under .pixi/envs/default/bin/python if not automatically selected, see documentation).