causal-forest
causal-forest copied to clipboard
Check for compatibility with sklearn functionals.
Change API to sklearn standard --even though treatment effect estimation is a mix between supervised and unsupervised learning and therefore any estimator can never utilize the features of sklearn it eases readability and consistency.
To do:
- Construct estimator class
cforest
which implements afit
andpredict
method. - Hide all other functionalities
- Check if there is any compatibility with sklearn.
In commit (d95a7cfa0cc019b10bd0738fbd186f4e73c81613) I implemented a wrapper class CausalForest
which implements the methods fit
, predict
, save
and load
.
What remains to be done:
- Check if I can hide private functions
- Check for compatibility with sklearn