MicrobeLab / DeepMicrobes

DeepMicrobes: taxonomic classification for metagenomics with deep learning
https://doi.org/10.1093/nargab/lqaa009
Apache License 2.0
81 stars 21 forks source link

About train_epoch #33

Open XLOXL opened 1 month ago

XLOXL commented 1 month ago

Hello, I would like to ask if the setting of train_epoch has an impact on the model. I see that you have set the default value to 1, but the common values for epochs are 50, 100, etc. I don’t quite understand

MicrobeLab commented 1 month ago

I generated a very large training set, and if the model did not converge, I generated more data rather than trained on the same dataset again. That's why the epoch is set to 1.

XLOXL commented 1 month ago

I see. Thank you very much for your reply!