plismbench.models package#

Unit tests for plismbench.models.

class plismbench.models.StringEnum(value)[source]#

Bases: Enum

A base class string enumerator.

classmethod choices()[source]#

Get Enum names.

class plismbench.models.FeatureExtractorsEnum(value)[source]#

Bases: StringEnum

A class enumerator for feature extractors.

H0_MINI = 'h0_mini'#
HOPTIMUS0 = 'hoptimus0'#
KAIKO_VIT_BASE = 'kaiko_vit_base'#
KAIKO_VIT_LARGE = 'kaiko_vit_large'#
MIDNIGHT_12K = 'midnight_12k'#
VIRCHOW = 'virchow'#
VIRCHOW2 = 'virchow2'#
PROVGIGAPATH = 'provgigapath'#
CONCH = 'conch'#
CONCHV15 = 'conchv15'#
UNI = 'uni'#
UNI2H = 'uni2h'#
HIBOU_BASE = 'hibou_base'#
HIBOU_LARGE = 'hibou_large'#
PHIKON = 'phikon'#
PHIKONV2 = 'phikonv2'#
GPFM = 'gpfm'#
PLIP = 'plip'#
LUNIT_VIT_SMALL_8 = 'lunit_vit_small_8'#
DINOV2_VIT_GIANT_IMAGENET = 'dinov2_vit_giant_imagenet'#
init(device: int | list[int] | None, mixed_precision: bool = True, **kwargs) → Extractor[source]#

Initialize the feature extractor. Mixed precision is set by default.

Submodules#