omicverse.fm.select_model

omicverse.fm.select_model(adata_path, task, prefer_zero_shot=True, max_vram_gb=None)

Select the best foundation model for a task and dataset.

Parameters:

• adata_path (str) – Path to .h5ad file.

• task (str) – Task type ('embed', 'annotate', 'integrate').

• prefer_zero_shot (bool) – Prefer models that don’t require fine-tuning.

• max_vram_gb (int, optional) – Maximum VRAM constraint.

Returns:

Recommended model with rationale and fallback options.

Return type:

dict