adelie
adelie copied to clipboard
scikit compatible learners
Hi James,
great work on this. Was wondering if you were planning to provide a scikit-learn style API for this package? Would largely involve writing init/fit/predict/score methods, which would then allow adelie to be used in a number of useful downstream tasks (e.g. what i'm interested in - nuisance function estimation for AIPW / Rlearner for ATE/CATE estimation in econml / doubleml). Celer is a closely related package that provides a scikit-style API.
Totally understand if this isn't a priority; I might take a crack at it and send you a PR in case you're open to contributions.