grf-labs / policytree

Policy learning via doubly robust empirical welfare maximization over trees
https://grf-labs.github.io/policytree/
MIT License
74 stars 15 forks source link

Add zero dim input check #130

Closed erikcs closed 2 years ago

erikcs commented 2 years ago

If a user fits on different folds and accidentally passes an empty index, policy_tree would crash, this prevents that very unlikely scenario with an error message instead.