jacobpennington / nems-lite

WIP refactor of lbhb/NEMS and lbhb/nems_db
GNU General Public License v3.0
1 stars 1 forks source link

option for Model as scikit-learn estimator (low priority) #22

Open jacobpennington opened 2 years ago

jacobpennington commented 2 years ago

Look into difficulty of wrapping a Model as a scikit-learn Estimator (see: https://scikit-learn.org/stable/developers/develop.html). If not too difficult, this would allow for using NEMS Models directly with a lot of useful scikit tools like sklearn.model_selection.cross_val_score without needing to re-implement everything from scratch.