** Code for Learning to Rank for Optimal Treatment Allocation Under Resource Constraints **
Code for Learning to Rank for Optimal Treatment Allocation Under Resource Constraints, published at AIStats 2024.
Instructions:
- Install required packages in
requirements.txt
- Set relevant hyperparameters in
run_models.py
. a.method = 1
runs a random forest splitting based on AUTOC, whereasmethod = 0
runs a baseline model that splits to maximize MSE. b. Other relevant hyperparameters are - Run
run_models.py
a. Models and results will be saved out in relevant directories, with unique identifiers based on relevant parameters. These can be adjusted as needed.