mlreco.models.layers.cluster_cnn.factories module

mlreco.models.layers.cluster_cnn.factories.backbone_dict()[source]

returns dictionary of clustering models

mlreco.models.layers.cluster_cnn.factories.gs_kernel_dict()[source]

Returns dictionary of kernel function models.

mlreco.models.layers.cluster_cnn.factories.gs_kernel_construct(cfg)[source]
mlreco.models.layers.cluster_cnn.factories.cluster_model_dict()[source]

Returns dictionary of implemented clustering layers.

mlreco.models.layers.cluster_cnn.factories.spice_loss_dict()[source]

Returns dictionary of various clustering losses with enhancements.

mlreco.models.layers.cluster_cnn.factories.backbone_construct(name)[source]
mlreco.models.layers.cluster_cnn.factories.cluster_model_construct(cfg, name)[source]
mlreco.models.layers.cluster_cnn.factories.spice_loss_construct(name)[source]