Official implementation of Contextual Feature Selection with Conditional Stochastic Gates .
If you find this package useful or inspirational, please cite our work as follows
@inproceedings{sristi2024contextual,
title={Contextual Feature Selection with Conditional Stochastic Gates},
author={Sristi, Ram Dyuthi and Lindenbaum, Ofir and Lifshitz, Shira and Lavzin, Maria and Schiller, Jackie and Mishne, Gal and Benisty, Hadas},
booktitle={International Conference on Machine Learning},
pages={46375--46392},
year={2024},
organization={PMLR}
}