devo_ml.modelmanager.engines

Engine code literals to identify the ML engines and infer their model file extensions.

ONNX = 'ONNX'

Extensions: .onnx

H2O = 'H2O'

Constant denoting H2O model.

Extensions: .zip

BIGML = 'BIGML'

Constant denoting BigML model.

Extensions: .json

CATBOOST = 'CATBOOST'

Constant denoting CatBoost model.

Extensions: .cmb

DT = 'DT'

Constant denoting Decision Tree model.

Extensions: .zip

IDA = 'IDA'

Constant denoting Iterated Distillation and Amplification model.

Extensions: .zip

MLSTATS = 'MLSTATS'

Constant denoting ML STATS model.

Extensions: -

MUA = 'MUA'

Constant denoting MUA model.

Extensions: .zip

UNICODE = 'UNICODE'

Constant denoting UNICODE model.

Extensions: .zip

WORKFLOWS = 'WORKFLOWS'

Constant denoting WORKFLOWS model.

Extensions: .json

get_engine_extensions(engine_code)

Returns file extensions associated with an engine represented by its code.

An empty list will be returned if it is an unknown engine code.

Parameters:

engine_code (str) – The code of the engine

Return type:

Sequence[str]

Returns:

The file extensions or an empty list. Extensions include the dot, e.g. .json, .zip …

get_default_engine_extension(engine_code)

Returns the file extension associated with an engine represented by its code.

An empty string will be returned if it is an unknown engine code.

Parameters:

engine_code (str) – The code of the engine

Return type:

str

Returns:

The extension associate with an engine or empty string. Extensions include the dot, e.g. .json, .zip …