rxn.metrics.classification_translation.maybe_classify_predictions
- rxn.metrics.classification_translation.maybe_classify_predictions(classification_model, retro_files, batch_size, gpu)[source]
Classify the reactions for determining the diversity metric.
Only executed if a classification model is available.
- Parameters
classification_model (
Optional[Path]) –retro_files (
RetroFiles) –batch_size (
int) –gpu (
bool) –
- Return type
None