ModelManager#

class sdgx.models.manager.ModelManager(*args, **kwargs)[source]#

Bases: Manager

_load_dir(module)#: Import all python files in a submodule.

_normalize_name(name: str) → str#

_registed_cls: dict[str, type[self.register_type]]#

hookspecs_model = <module 'sdgx.models.extension' from '/home/docs/checkouts/readthedocs.org/user_builds/synthetic-data-generator/envs/latest/lib/python3.10/site-packages/sdgx/models/extension.py'>#: Hook specification model(where @hookspec is defined)

init(c, **kwargs: dict[str, Any])#

Init a new subclass of self.register_type.

Raises:

NotFoundError – if cls_name is not registered
InitializationError – if failed to initialize

init_model(model_name, **kwargs: dict[str, Any]) → SynthesizerModel[source]#: redirect to init

load(model: type[SynthesizerModel] | str, model_path, **kwargs) → SynthesizerModel[source]#

load_all_local_model()[source]#: Implement this function to load all local model

project_name: str = 'sdgx.model'#: Name of entry-point for extension

property registed_cls: dict[str, type]#

Access all registed class.

Lazy load, only load once.

property registed_models#: redirect to registed_cls

register(cls_name, cls: type)#: Register a new model, if the model is already registed, skip it.

register_type#: alias of SynthesizerModel