search

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

Model weights for training #30

Open Gayathri142 opened 8 months ago

Gayathri142 commented 8 months ago

Hi, Could you please elaborate a bit on the parameter '--model_dir', in the training part of the tool? Are there weights that come with the study, or do we have to generate them ourselves? If so, could you please guide me on how to generate the weights and add it to the model while training? Thanks, Gayathri

MicrobeLab commented 8 months ago

Hi, The value of --model_dir sets the folder to store weights during training and the folder from which trained weights are loaded in prediction. So, the folder should be same for training and testing. In training mode, the model will store the trained weights to this folder.

Gayathri142 commented 6 months ago

Thank you for your explanation.