Hi @lewtun my classes aren’t balanced but i’m using a sample to get an even 20% per class in each batch.
Thanks for the suggestions, I will try and implement a simple baseline and use cross-validation!