`neural_compressor.ux.components.model.repository`¶

Model repository.

Module Contents¶

Classes¶

ModelRepository

Model repository.

class neural_compressor.ux.components.model.repository.ModelRepository¶

Model repository.

get_model(path: str) → neural_compressor.ux.components.model.model.Model¶: Get Model for given path.

get_frameworks() → List[str]¶: Get list of supported frameworks.

static get_framework_from_path(model_path: str) → str¶

Get framework name from model extension.

Parameters:: model_path – Path to model.

static get_supported_frameworks() → List[str]¶: Get list of supported frameworks.

static is_model_path(path: str) → bool¶: Check if provided path is for model.