search

naplab / naplib-python

Tools and functions for neural data processing and analysis in python
https://naplib-python.readthedocs.io/en/latest/index.html
MIT License
21 stars 9 forks source link

Change TRF fitting internal cross-validation to use naplib.model_selection.KFold #8

Closed gavinmischler closed 2 years ago

gavinmischler commented 2 years ago

Internally, TRF performs cross-validation during fitting, but it uses a custom cross-validation splitter, so just convert this to naplib.model_selection.KFold to improve code re-use.

gavinmischler commented 2 years ago

Resolved by #70 which updated TRF to allow sklearn linear_model instances (including models which include cross-validation internally, like RidgeCV)