Reproduce training

Hello,

Could you please document in the repository how to run the training in order to reproduce the paper results?

You only documented how to run

python Predict.py dataset file

It would nice to see how to train the models in order to obtain the paper results.

Thanks.

P.S.

I was doing some experiments, trying to reproduce the paper results and I saw this line:

ERGO-II/Trainer.py

Line 235 in 7a25b90

with open('Samples/' + self.dataset + '_test_samples.pickle', 'rb') as handle:

It seems you do validation on the test set.
Do you have an additional left-out test set, on which you computed the paper results? Or are the results in the paper computed on the validation set?

Thanks.