Groupkfold or Groupshufflesplit Cross Validation

aerdem4 / lofo-importance

Leave One Feature Out Importance

MIT License

810 stars 83 forks source link

Groupkfold or Groupshufflesplit Cross Validation #37

Closed kmedved closed 2 years ago

kmedved commented 2 years ago

Thanks for the very cool package. I was wondering if it is possible to pass a group_id somehow in order to do Groupkfold or Groupshufflesplit cross validation? I don't see anywhere obvious, but wanted to check in case I'm missing anything.

Thanks.

aerdem4 commented 2 years ago

@kmedved cv parameter supports any custom iterable (list of (train_ind, val_ind)). I suggested a workaround on #28 Please let me know if this workaround works for you.