yandex-research / rtdl-revisiting-models

(NeurIPS 2021) Revisiting Deep Learning Models for Tabular Data
https://arxiv.org/abs/2106.11959
Apache License 2.0
204 stars 42 forks source link

k-fold cross validation #30

Closed manmeet3591 closed 2 years ago

manmeet3591 commented 2 years ago

Is there a way to perform k-fold cross validation with this setup?

Yura52 commented 2 years ago

If you don't need hyperparameter tuning, then you can store k possible splits as k separate datasets, and then average the metrics for all of them.

However, if you need hyperparameter tuning, then you have two options:

Does this help?

manmeet3591 commented 2 years ago

Thank you. Yes, this definitely helps.