plismbench.models.bioptimus module

Models from Bioptimus company.

class plismbench.models.bioptimus.H0Mini(device: int | list[int] | None = -1, mixed_precision: bool = False)

Bases: Extractor

H0-mini model developped by Owkin & Bioptimus available on Hugging-Face (1).

You will need to be granted access to be able to use this model.

Note

1. https://huggingface.co/bioptimus/H0-mini

Parameters:

• device (int | list[int] | None = DEFAULT_DEVICE,) – Compute resources to use. If None, will use all available GPUs. If -1, extraction will run on CPU.

• mixed_precision (bool = True) – Whether to use mixed_precision.

property transform: Compose

Transform method to apply element wise.