mendevi.cst.model¶

All the available models.

Functions

import_all_models()

Import all the class in mendevi/models/*.py that inherit from Model.

Details

mendevi.cst.model.import_all_models() → dict[str, Model][source]

Import all the class in mendevi/models/*.py that inherit from Model.

Examples¶

>>> from pprint import pprint
>>> from mendevi.cst.model import import_all_models
>>> pprint(import_all_models())
{'DNN': <class 'mendevi.models.nn.DNN'>,
 'DecodeEnergyNN': <class 'mendevi.models.nn.DecodeEnergyNN'>,
 'EncodeEnergyNN': <class 'mendevi.models.nn.EncodeEnergyNN'>,
 'EncodeLinear': <class 'mendevi.models.lr.EncodeLinear'>,
 'GPR': <class 'mendevi.models.gpr.GPR'>,
 'KernelMultiHead': <class 'mendevi.models.nn.KernelMultiHead'>,
 'LR': <class 'mendevi.models.lr.LR'>,
 'NearestL2': <class 'mendevi.models.interp.NearestL2'>,
 'PowerCores': <class 'mendevi.models.power_cores.PowerCores'>,
 'QBP': <class 'mendevi.models.qbp.QBP'>,
 'QualityNN': <class 'mendevi.models.nn.QualityNN'>}
>>>