search

pytorch-ignite / examples

Examples, tutorials, and how-to guides
BSD 3-Clause "New" or "Revised" License
11 stars 12 forks source link

Cross validation guide #28

Closed Priyansi closed 3 years ago

Priyansi commented 3 years ago

Ideas on how to do cross-validation: https://github.com/pytorch/ignite/issues/1384

LiviaCavalcanti commented 3 years ago

Hi! Is it ok to use scikit-learn’s k-fold functionality to split the data?

vfdev-5 commented 3 years ago

@LiviaCavalcanti yes, sure!

Priyansi commented 3 years ago

Hey @LiviaCavalcanti , I started working on this notebook last week. It's a basic code sample. It still needs some explanation and a bit of restructuring. Do you want to work on this notebook and send a PR for this?

For ideas on how to add explanation: refer to https://pytorch-ignite.ai/how-to-guides/

Please feel free to ask any more questions!

LiviaCavalcanti commented 3 years ago

Wow you've done all the hard work. I would be glad to work on the explanation! I will send the PR with it.

Priyansi commented 3 years ago

Thank you @LiviaCavalcanti ! BTW, the code is just an example. If you feel like it can be simplified, feel free to make changes there too :)