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
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions