mlreco.models.factories module

mlreco.models.factories.model_dict()[source]

Returns dictionary of model classes using name keys (strings).

Returns

Return type

dict

mlreco.models.factories.construct(name)[source]

Returns an instance of a model class based on its name key (string).

Parameters

name (str) – Key for the model. See source code for list of available models.

Returns

Return type

object