Skip to content

differentiate between available and used models #18

@sfluegel05

Description

@sfluegel05

Use case

Someone wants to unselect a model for a given prediction (without restarting the whole ensemble). This is especially common for our frontend which has toggle buttons.

Current status

To do this, one has to do something like

ensemble = BaseEnsemble(my_config)
full_preds = predict_smiles_list(my_smiles)
all_models = ensemble.models
ensemble.models = ensemble.models[:-1] # throw out last models
partial_preds = predict_smiles_list(my_smiles)
ensemble.models = all_models # restore models

This is cumbersome and error-prone.

How it should be

ensemble = BaseEnsemble(my_config)
full_preds = predict_smiles_list(my_smiles)
partial_preds = predict_smiles_list(my_smiles, use_models=["Model1", "Model2", "Model3")

This is easy to use (given that the user knows their model names, but that can be easily inferred from the ensemble. use_models specifies which models to use (default: all that are available),

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions