Index

_

__dict__ (analysis.classes.ui.FullChainPredictor attribute) (mlreco.iotools.collates.VolumeBoundaries attribute) (mlreco.main_funcs.Handlers attribute) (mlreco.trainval.trainval attribute) (mlreco.utils.utils.ChunkCSVData attribute) (mlreco.utils.utils.CSVData attribute) (mlreco.utils.utils.ForwardData attribute) (mlreco.utils.utils.stopwatch attribute) __getitem__() (mlreco.iotools.datasets.LArCVDataset method) (mlreco.utils.utils.ForwardData method) __init__() (analysis.classes.ui.FullChainEvaluator method) (analysis.classes.ui.FullChainPredictor method) (mlreco.iotools.collates.VolumeBoundaries method) (mlreco.iotools.datasets.LArCVDataset method) (mlreco.iotools.samplers.AbstractBatchSampler method) (mlreco.models.bayes_uresnet.BayesianUResNet method) (mlreco.models.bayes_uresnet.DUQSegmentationLoss method) (mlreco.models.bayes_uresnet.DUQUResNet method) (mlreco.models.bayes_uresnet.SegmentationLoss method) (mlreco.models.full_chain.FullChain method) (mlreco.models.full_chain.FullChainLoss method) (mlreco.models.graph_spice.GraphSPICELoss method) (mlreco.models.graph_spice.MinkGraphSPICE method) (mlreco.models.grappa.GNN method) (mlreco.models.grappa.GNNLoss method) (mlreco.models.layers.cluster_cnn.embeddings.Attention method) (mlreco.models.layers.cluster_cnn.embeddings.ExpandAs method) (mlreco.models.layers.cluster_cnn.embeddings.SPICE method) (mlreco.models.layers.cluster_cnn.graph_spice_embedder.GraphSPICEEmbedder method) (mlreco.models.layers.cluster_cnn.gs_kernels.BilinearKernel method) (mlreco.models.layers.cluster_cnn.gs_kernels.BilinearNNKernel method) (mlreco.models.layers.cluster_cnn.gs_kernels.DefaultKernel method) (mlreco.models.layers.cluster_cnn.gs_kernels.MixedKernel method) (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.GraphSPICEEmbeddingLoss method) (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.NodeEdgeHybridLoss method) (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.WeightedEdgeLoss method) (mlreco.models.layers.cluster_cnn.losses.lovasz.StableBCELoss method) (mlreco.models.layers.cluster_cnn.losses.misc.BinaryCELogDiceLoss method) (mlreco.models.layers.cluster_cnn.losses.misc.BinaryLogDiceLoss method) (mlreco.models.layers.cluster_cnn.losses.misc.FocalLoss method) (mlreco.models.layers.cluster_cnn.losses.misc.IoUScore method) (mlreco.models.layers.cluster_cnn.losses.misc.LovaszHingeLoss method) (mlreco.models.layers.cluster_cnn.losses.misc.LovaszSoftmaxWithLogitsLoss method) (mlreco.models.layers.cluster_cnn.losses.misc.MincutLoss method) (mlreco.models.layers.cluster_cnn.losses.misc.VectorEstimationLoss method) (mlreco.models.layers.cluster_cnn.losses.misc.WeightedFocalLoss method) (mlreco.models.layers.cluster_cnn.losses.radius_nnloss.DensityBasedNNLoss method) (mlreco.models.layers.cluster_cnn.losses.single_layers.DiscriminativeLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.CELovaszLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELoss2 method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELossBivariate method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskFocalLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszHingeLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterBC method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterLoss2 method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MultiVariateLovasz method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast.SPICEInterLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast.SPICELoss method) (mlreco.models.layers.common.blocks.ASPP method) (mlreco.models.layers.common.blocks.AtrousIIBlock method) (mlreco.models.layers.common.blocks.CascadeDilationBlock method) (mlreco.models.layers.common.blocks.ConvolutionBlock method) (mlreco.models.layers.common.blocks.DenseResBlock method) (mlreco.models.layers.common.blocks.DropoutBlock method) (mlreco.models.layers.common.blocks.Identity method) (mlreco.models.layers.common.blocks.MBConv method) (mlreco.models.layers.common.blocks.MBResConv method) (mlreco.models.layers.common.blocks.MBResConvSE method) (mlreco.models.layers.common.blocks.ResNetBlock method) (mlreco.models.layers.common.blocks.ResNeXtBlock method) (mlreco.models.layers.common.blocks.SEBlock method) (mlreco.models.layers.common.blocks.SEResNetBlock method) (mlreco.models.layers.common.blocks.SparseToDense method) (mlreco.models.layers.common.blocks.SPP method) (mlreco.models.layers.common.cluster_pool.ClusterPool method) (mlreco.models.layers.common.cnn_encoder.SparseResidualEncoder method) (mlreco.models.layers.common.dbscan.DBSCANFragmenter method) (mlreco.models.layers.common.extract_feature_map.MinkGhostMask method) (mlreco.models.layers.common.fpn.FPN method) (mlreco.models.layers.common.gnn_full_chain.FullChainGNN method) (mlreco.models.layers.common.gnn_full_chain.FullChainLoss method) (mlreco.models.layers.common.mobilenet.MB3Encoder method) (mlreco.models.layers.common.mobilenet.MobileNetV3 method) (mlreco.models.layers.common.momentum.DeepVertexNet method) (mlreco.models.layers.common.momentum.EvidentialMomentumNet method) (mlreco.models.layers.common.momentum.MomentumNet method) (mlreco.models.layers.common.momentum.VertexNet method) (mlreco.models.layers.common.nonlinearities.MinkowskiMish method) (mlreco.models.layers.common.normalizations.MinkowskiAdaIN method) (mlreco.models.layers.common.normalizations.MinkowskiPixelNorm method) (mlreco.models.layers.common.ppnplus.AttentionMask method) (mlreco.models.layers.common.ppnplus.ExpandAs method) (mlreco.models.layers.common.ppnplus.MergeConcat method) (mlreco.models.layers.common.ppnplus.PPN method) (mlreco.models.layers.common.ppnplus.PPNLonelyLoss method) (mlreco.models.layers.common.senet.SENet method) (mlreco.models.layers.common.sparse_generator.SparseGenerator method) (mlreco.models.layers.common.sparse_generator.SparseGeneratorSimple method) (mlreco.models.layers.common.uresnet_layers.UResNet method) (mlreco.models.layers.common.uresnet_layers.UResNetDecoder method) (mlreco.models.layers.common.uresnet_layers.UResNetEncoder method) (mlreco.models.layers.common.uresnext.UResNeXt method) (mlreco.models.layers.gnn.encoders.cnn.ClustCNNMinkEdgeEncoder method) (mlreco.models.layers.gnn.encoders.cnn.ClustCNNMinkNodeEncoder method) (mlreco.models.layers.gnn.encoders.geometric.ClustGeoEdgeEncoder method) (mlreco.models.layers.gnn.encoders.geometric.ClustGeoNodeEncoder method) (mlreco.models.layers.gnn.encoders.mixed.ClustMixEdgeEncoder method) (mlreco.models.layers.gnn.encoders.mixed.ClustMixNodeEncoder method) (mlreco.models.layers.gnn.losses.edge_channel.EdgeChannelLoss method) (mlreco.models.layers.gnn.losses.node_grouping.GNNGroupingLoss method) (mlreco.models.layers.gnn.losses.node_kinematics.BerHuLoss method) (mlreco.models.layers.gnn.losses.node_kinematics.LogRMSE method) (mlreco.models.layers.gnn.losses.node_kinematics.NodeEvidentialKinematicsLoss method) (mlreco.models.layers.gnn.losses.node_kinematics.NodeKinematicsLoss method) (mlreco.models.layers.gnn.losses.node_kinematics.NodeTransformerLoss method) (mlreco.models.layers.gnn.losses.node_primary.NodePrimaryLoss method) (mlreco.models.layers.gnn.losses.node_type.NodeTypeLoss method) (mlreco.models.layers.gnn.message_passing.agnnconv.AGNNConvModel method) (mlreco.models.layers.gnn.message_passing.agnnconv.EdgeLayer method) (mlreco.models.layers.gnn.message_passing.agnnconv.GlobalModel method) (mlreco.models.layers.gnn.message_passing.agnnconv.NodeLayer method) (mlreco.models.layers.gnn.message_passing.dgtransformer.DeepGraphTransformer method) (mlreco.models.layers.gnn.message_passing.econv.EConvModel method) (mlreco.models.layers.gnn.message_passing.econv.EdgeLayer method) (mlreco.models.layers.gnn.message_passing.econv.GlobalModel method) (mlreco.models.layers.gnn.message_passing.econv.NodeLayer method) (mlreco.models.layers.gnn.message_passing.gatconv.EdgeLayer method) (mlreco.models.layers.gnn.message_passing.gatconv.GATConvModel method) (mlreco.models.layers.gnn.message_passing.gatconv.GlobalModel method) (mlreco.models.layers.gnn.message_passing.gatconv.NodeLayer method) (mlreco.models.layers.gnn.message_passing.meta.EdgeLayer method) (mlreco.models.layers.gnn.message_passing.meta.GlobalModel method) (mlreco.models.layers.gnn.message_passing.meta.MetaLayerModel method) (mlreco.models.layers.gnn.message_passing.meta.NodeLayer method) (mlreco.models.layers.gnn.message_passing.nnconv.EdgeLayer method) (mlreco.models.layers.gnn.message_passing.nnconv.GlobalModel method) (mlreco.models.layers.gnn.message_passing.nnconv.NNConvModel method) (mlreco.models.layers.gnn.message_passing.nnconv.NodeLayer method) (mlreco.models.layers.gnn.message_passing.nnconv_elu.EdgeLayer method) (mlreco.models.layers.gnn.message_passing.nnconv_elu.GlobalModel method) (mlreco.models.layers.gnn.message_passing.nnconv_elu.NNConvModel method) (mlreco.models.layers.gnn.message_passing.nnconv_elu.NodeLayer method) (mlreco.models.layers.gnn.message_passing.nnconv_old.EdgeLayer method) (mlreco.models.layers.gnn.message_passing.nnconv_old.GlobalModel method) (mlreco.models.layers.gnn.message_passing.nnconv_old.NNConvModel method) (mlreco.models.layers.gnn.message_passing.nnconv_old.NodeLayer method) (mlreco.models.layers.gnn.normalizations.BatchNorm method) (mlreco.models.layers.gnn.normalizations.InstanceNorm method) (mlreco.models.singlep.BayesianParticleClassifier method) (mlreco.models.singlep.DUQParticleClassifier method) (mlreco.models.singlep.EvidentialLearningLoss method) (mlreco.models.singlep.EvidentialParticleClassifier method) (mlreco.models.singlep.MultiLabelCrossEntropy method) (mlreco.models.singlep.ParticleImageClassifier method) (mlreco.models.singlep.ParticleTypeLoss method) (mlreco.models.spice.MinkSPICE method) (mlreco.models.spice.SPICELoss method) (mlreco.models.uresnet.SegmentationLoss method) (mlreco.models.uresnet.UResNet_Chain method) (mlreco.models.uresnet_ppn_chain.UResNetPPN method) (mlreco.models.uresnet_ppn_chain.UResNetPPNLoss method) (mlreco.trainval.trainval method) (mlreco.utils.adabound.AdaBound method) (mlreco.utils.adabound.AdaBoundW method) (mlreco.utils.data_parallel.DataParallel method) (mlreco.utils.utils.ChunkCSVData method) (mlreco.utils.utils.CSVData method) (mlreco.utils.utils.ForwardData method) (mlreco.utils.utils.stopwatch method) __iter__() (mlreco.iotools.samplers.BootstrapBatchSampler method) (mlreco.iotools.samplers.RandomSequenceSampler method) (mlreco.iotools.samplers.SequentialBatchSampler method) __len__() (mlreco.iotools.datasets.LArCVDataset method) (mlreco.iotools.samplers.AbstractBatchSampler method) __module__ (analysis.classes.ui.FullChainEvaluator attribute) (analysis.classes.ui.FullChainPredictor attribute) (mlreco.iotools.collates.VolumeBoundaries attribute) (mlreco.iotools.datasets.LArCVDataset attribute) (mlreco.iotools.samplers.AbstractBatchSampler attribute) (mlreco.iotools.samplers.BootstrapBatchSampler attribute) (mlreco.iotools.samplers.RandomSequenceSampler attribute) (mlreco.iotools.samplers.SequentialBatchSampler attribute) (mlreco.main_funcs.Handlers attribute) (mlreco.models.bayes_uresnet.BayesianUResNet attribute) (mlreco.models.bayes_uresnet.DUQSegmentationLoss attribute) (mlreco.models.bayes_uresnet.DUQUResNet attribute) (mlreco.models.bayes_uresnet.SegmentationLoss attribute) (mlreco.models.full_chain.FullChain attribute) (mlreco.models.full_chain.FullChainLoss attribute) (mlreco.models.graph_spice.GraphSPICELoss attribute) (mlreco.models.graph_spice.MinkGraphSPICE attribute) (mlreco.models.grappa.GNN attribute) (mlreco.models.grappa.GNNLoss attribute) (mlreco.models.layers.cluster_cnn.embeddings.Attention attribute) (mlreco.models.layers.cluster_cnn.embeddings.ExpandAs attribute) (mlreco.models.layers.cluster_cnn.embeddings.SPICE attribute) (mlreco.models.layers.cluster_cnn.graph_spice_embedder.GraphSPICEEmbedder attribute) (mlreco.models.layers.cluster_cnn.gs_kernels.BilinearKernel attribute) (mlreco.models.layers.cluster_cnn.gs_kernels.BilinearNNKernel attribute) (mlreco.models.layers.cluster_cnn.gs_kernels.DefaultKernel attribute) (mlreco.models.layers.cluster_cnn.gs_kernels.MixedKernel attribute) (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.GraphSPICEEmbeddingLoss attribute) (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.NodeEdgeHybridLoss attribute) (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.WeightedEdgeLoss attribute) (mlreco.models.layers.cluster_cnn.losses.lovasz.StableBCELoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.BinaryCELogDiceLoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.BinaryLogDiceLoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.FocalLoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.IoUScore attribute) (mlreco.models.layers.cluster_cnn.losses.misc.LovaszHingeLoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.LovaszSoftmaxWithLogitsLoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.MincutLoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.VectorEstimationLoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.WeightedFocalLoss attribute) (mlreco.models.layers.cluster_cnn.losses.radius_nnloss.DensityBasedNNLoss attribute) (mlreco.models.layers.cluster_cnn.losses.single_layers.DiscriminativeLoss attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.CELovaszLoss attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELoss attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELoss2 attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELossBivariate attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskFocalLoss attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszHingeLoss attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterBC attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterLoss attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterLoss2 attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MultiVariateLovasz attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast.SPICEInterLoss attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast.SPICELoss attribute) (mlreco.models.layers.common.blocks.ASPP attribute) (mlreco.models.layers.common.blocks.AtrousIIBlock attribute) (mlreco.models.layers.common.blocks.CascadeDilationBlock attribute) (mlreco.models.layers.common.blocks.ConvolutionBlock attribute) (mlreco.models.layers.common.blocks.DenseResBlock attribute) (mlreco.models.layers.common.blocks.DropoutBlock attribute) (mlreco.models.layers.common.blocks.Identity attribute) (mlreco.models.layers.common.blocks.MBConv attribute) (mlreco.models.layers.common.blocks.MBResConv attribute) (mlreco.models.layers.common.blocks.MBResConvSE attribute) (mlreco.models.layers.common.blocks.ResNetBlock attribute) (mlreco.models.layers.common.blocks.ResNeXtBlock attribute) (mlreco.models.layers.common.blocks.SEBlock attribute) (mlreco.models.layers.common.blocks.SEResNetBlock attribute) (mlreco.models.layers.common.blocks.SparseToDense attribute) (mlreco.models.layers.common.blocks.SPP attribute) (mlreco.models.layers.common.cluster_pool.ClusterPool attribute) (mlreco.models.layers.common.cnn_encoder.SparseResidualEncoder attribute) (mlreco.models.layers.common.dbscan.DBSCANFragmenter attribute) (mlreco.models.layers.common.extract_feature_map.MinkGhostMask attribute) (mlreco.models.layers.common.fpn.FPN attribute) (mlreco.models.layers.common.gnn_full_chain.FullChainGNN attribute) (mlreco.models.layers.common.gnn_full_chain.FullChainLoss attribute) (mlreco.models.layers.common.mobilenet.MB3Encoder attribute) (mlreco.models.layers.common.mobilenet.MobileNetV3 attribute) (mlreco.models.layers.common.momentum.DeepVertexNet attribute) (mlreco.models.layers.common.momentum.EvidentialMomentumNet attribute) (mlreco.models.layers.common.momentum.MomentumNet attribute) (mlreco.models.layers.common.momentum.VertexNet attribute) (mlreco.models.layers.common.nonlinearities.MinkowskiELU attribute) (mlreco.models.layers.common.nonlinearities.MinkowskiLeakyReLU attribute) (mlreco.models.layers.common.nonlinearities.MinkowskiMish attribute) (mlreco.models.layers.common.normalizations.MinkowskiAdaIN attribute) (mlreco.models.layers.common.normalizations.MinkowskiGroupNorm attribute) (mlreco.models.layers.common.normalizations.MinkowskiPixelNorm attribute) (mlreco.models.layers.common.ppnplus.AttentionMask attribute) (mlreco.models.layers.common.ppnplus.ExpandAs attribute) (mlreco.models.layers.common.ppnplus.MergeConcat attribute) (mlreco.models.layers.common.ppnplus.PPN attribute) (mlreco.models.layers.common.ppnplus.PPNLonelyLoss attribute) (mlreco.models.layers.common.senet.SENet attribute) (mlreco.models.layers.common.sparse_generator.SparseGenerator attribute) (mlreco.models.layers.common.sparse_generator.SparseGeneratorSimple attribute) (mlreco.models.layers.common.uresnet_layers.UResNet attribute) (mlreco.models.layers.common.uresnet_layers.UResNetDecoder attribute) (mlreco.models.layers.common.uresnet_layers.UResNetEncoder attribute) (mlreco.models.layers.common.uresnext.UResNeXt attribute) (mlreco.models.layers.gnn.encoders.cnn.ClustCNNMinkEdgeEncoder attribute) (mlreco.models.layers.gnn.encoders.cnn.ClustCNNMinkNodeEncoder attribute) (mlreco.models.layers.gnn.encoders.geometric.ClustGeoEdgeEncoder attribute) (mlreco.models.layers.gnn.encoders.geometric.ClustGeoNodeEncoder attribute) (mlreco.models.layers.gnn.encoders.mixed.ClustMixEdgeEncoder attribute) (mlreco.models.layers.gnn.encoders.mixed.ClustMixNodeEncoder attribute) (mlreco.models.layers.gnn.losses.edge_channel.EdgeChannelLoss attribute) (mlreco.models.layers.gnn.losses.node_grouping.GNNGroupingLoss attribute) (mlreco.models.layers.gnn.losses.node_kinematics.BerHuLoss attribute) (mlreco.models.layers.gnn.losses.node_kinematics.LogRMSE attribute) (mlreco.models.layers.gnn.losses.node_kinematics.NodeEvidentialKinematicsLoss attribute) (mlreco.models.layers.gnn.losses.node_kinematics.NodeKinematicsLoss attribute) (mlreco.models.layers.gnn.losses.node_kinematics.NodeTransformerLoss attribute) (mlreco.models.layers.gnn.losses.node_primary.NodePrimaryLoss attribute) (mlreco.models.layers.gnn.losses.node_type.NodeTypeLoss attribute) (mlreco.models.layers.gnn.message_passing.agnnconv.AGNNConvModel attribute) (mlreco.models.layers.gnn.message_passing.agnnconv.EdgeLayer attribute) (mlreco.models.layers.gnn.message_passing.agnnconv.GlobalModel attribute) (mlreco.models.layers.gnn.message_passing.agnnconv.NodeLayer attribute) (mlreco.models.layers.gnn.message_passing.dgtransformer.DeepGraphTransformer attribute) (mlreco.models.layers.gnn.message_passing.econv.EConvModel attribute) (mlreco.models.layers.gnn.message_passing.econv.EdgeLayer attribute) (mlreco.models.layers.gnn.message_passing.econv.GlobalModel attribute) (mlreco.models.layers.gnn.message_passing.econv.NodeLayer attribute) (mlreco.models.layers.gnn.message_passing.gatconv.EdgeLayer attribute) (mlreco.models.layers.gnn.message_passing.gatconv.GATConvModel attribute) (mlreco.models.layers.gnn.message_passing.gatconv.GlobalModel attribute) (mlreco.models.layers.gnn.message_passing.gatconv.NodeLayer attribute) (mlreco.models.layers.gnn.message_passing.meta.EdgeLayer attribute) (mlreco.models.layers.gnn.message_passing.meta.GlobalModel attribute) (mlreco.models.layers.gnn.message_passing.meta.MetaLayerModel attribute) (mlreco.models.layers.gnn.message_passing.meta.NodeLayer attribute) (mlreco.models.layers.gnn.message_passing.nnconv.EdgeLayer attribute) (mlreco.models.layers.gnn.message_passing.nnconv.GlobalModel attribute) (mlreco.models.layers.gnn.message_passing.nnconv.NNConvModel attribute) (mlreco.models.layers.gnn.message_passing.nnconv.NodeLayer attribute) (mlreco.models.layers.gnn.message_passing.nnconv_elu.EdgeLayer attribute) (mlreco.models.layers.gnn.message_passing.nnconv_elu.GlobalModel attribute) (mlreco.models.layers.gnn.message_passing.nnconv_elu.NNConvModel attribute) (mlreco.models.layers.gnn.message_passing.nnconv_elu.NodeLayer attribute) (mlreco.models.layers.gnn.message_passing.nnconv_old.EdgeLayer attribute) (mlreco.models.layers.gnn.message_passing.nnconv_old.GlobalModel attribute) (mlreco.models.layers.gnn.message_passing.nnconv_old.NNConvModel attribute) (mlreco.models.layers.gnn.message_passing.nnconv_old.NodeLayer attribute) (mlreco.models.layers.gnn.normalizations.BatchNorm attribute) (mlreco.models.layers.gnn.normalizations.InstanceNorm attribute) (mlreco.models.singlep.BayesianParticleClassifier attribute) (mlreco.models.singlep.DUQParticleClassifier attribute) (mlreco.models.singlep.EvidentialLearningLoss attribute) (mlreco.models.singlep.EvidentialParticleClassifier attribute) (mlreco.models.singlep.MultiLabelCrossEntropy attribute) (mlreco.models.singlep.ParticleImageClassifier attribute) (mlreco.models.singlep.ParticleTypeLoss attribute) (mlreco.models.spice.MinkSPICE attribute) (mlreco.models.spice.SPICELoss attribute) (mlreco.models.uresnet.SegmentationLoss attribute) (mlreco.models.uresnet.UResNet_Chain attribute) (mlreco.models.uresnet_ppn_chain.UResNetPPN attribute) (mlreco.models.uresnet_ppn_chain.UResNetPPNLoss attribute) (mlreco.trainval.trainval attribute) (mlreco.utils.adabound.AdaBound attribute) (mlreco.utils.adabound.AdaBoundW attribute) (mlreco.utils.data_parallel.DataParallel attribute) (mlreco.utils.utils.ChunkCSVData attribute) (mlreco.utils.utils.CSVData attribute) (mlreco.utils.utils.ForwardData attribute) (mlreco.utils.utils.stopwatch attribute)

__orig_bases__ (mlreco.models.layers.common.blocks.ASPP attribute) (mlreco.models.layers.common.blocks.AtrousIIBlock attribute) (mlreco.models.layers.common.blocks.CascadeDilationBlock attribute) (mlreco.models.layers.common.blocks.ConvolutionBlock attribute) (mlreco.models.layers.common.blocks.DropoutBlock attribute) (mlreco.models.layers.common.blocks.MBConv attribute) (mlreco.models.layers.common.blocks.MBResConv attribute) (mlreco.models.layers.common.blocks.MBResConvSE attribute) (mlreco.models.layers.common.blocks.ResNetBlock attribute) (mlreco.models.layers.common.blocks.ResNeXtBlock attribute) (mlreco.models.layers.common.blocks.SEBlock attribute) (mlreco.models.layers.common.blocks.SEResNetBlock attribute) (mlreco.models.layers.common.blocks.SPP attribute) (mlreco.models.layers.common.nonlinearities.MinkowskiELU attribute) (mlreco.models.layers.common.nonlinearities.MinkowskiLeakyReLU attribute) __parameters__ (mlreco.iotools.datasets.LArCVDataset attribute) (mlreco.iotools.samplers.AbstractBatchSampler attribute) (mlreco.iotools.samplers.BootstrapBatchSampler attribute) (mlreco.iotools.samplers.RandomSequenceSampler attribute) (mlreco.iotools.samplers.SequentialBatchSampler attribute) __repr__() (analysis.classes.ui.FullChainPredictor method) (mlreco.models.layers.common.nonlinearities.MinkowskiLeakyReLU method) (mlreco.models.layers.common.nonlinearities.MinkowskiMish method) (mlreco.models.layers.common.normalizations.MinkowskiPixelNorm method) (mlreco.models.layers.gnn.normalizations.BatchNorm method) (mlreco.utils.utils.ForwardData method) __setstate__() (mlreco.utils.adabound.AdaBound method) (mlreco.utils.adabound.AdaBoundW method) __weakref__ (analysis.classes.ui.FullChainPredictor attribute) (mlreco.iotools.collates.VolumeBoundaries attribute) (mlreco.main_funcs.Handlers attribute) (mlreco.trainval.trainval attribute) (mlreco.utils.utils.ChunkCSVData attribute) (mlreco.utils.utils.CSVData attribute) (mlreco.utils.utils.ForwardData attribute) (mlreco.utils.utils.stopwatch attribute) _check_volume() (analysis.classes.ui.FullChainPredictor method) _fit_predict_fragments() (analysis.classes.ui.FullChainPredictor method) _fit_predict_groups() (analysis.classes.ui.FullChainPredictor method) _fit_predict_gspice_fragments() (analysis.classes.ui.FullChainPredictor method) _fit_predict_interaction_labels() (analysis.classes.ui.FullChainPredictor method) _fit_predict_pids() (analysis.classes.ui.FullChainPredictor method) _fit_predict_ppn() (analysis.classes.ui.FullChainPredictor method) _fit_predict_semantics() (analysis.classes.ui.FullChainPredictor method) _fit_predict_vertex_info() (analysis.classes.ui.FullChainPredictor method) _forward() (mlreco.trainval.trainval method) _get_entries() (analysis.classes.ui.FullChainPredictor method) _run_flash_matching() (analysis.classes.ui.FullChainPredictor method) _translate() (analysis.classes.ui.FullChainPredictor method) _untranslate() (analysis.classes.ui.FullChainPredictor method)

A

AbstractBatchSampler (class in mlreco.iotools.samplers) activations_construct() (in module mlreco.models.layers.common.activation_normalization_factories) activations_dict() (in module mlreco.models.layers.common.activation_normalization_factories) AdaBound (class in mlreco.utils.adabound) AdaBoundW (class in mlreco.utils.adabound) adapt_labels() (in module mlreco.utils.deghosting) adapt_labels_knn() (in module mlreco.utils.deghosting) adapt_labels_numpy() (in module mlreco.utils.deghosting) affine (mlreco.models.layers.gnn.normalizations.BatchNorm attribute) (mlreco.models.layers.gnn.normalizations.InstanceNorm attribute) AGNNConvModel (class in mlreco.models.layers.gnn.message_passing.agnnconv) all_nb() (in module mlreco.utils.numba) AMI() (in module mlreco.utils.metrics) analysis module analysis.algorithms module analysis.algorithms.calorimetry module analysis.algorithms.point_matching module analysis.algorithms.selections module analysis.algorithms.selections.michel_electrons module

analysis.algorithms.selections.stopping_muons module analysis.algorithms.selections.through_going_muons module analysis.algorithms.utils module analysis.classes module analysis.classes.particle module analysis.classes.ui module analysis.decorator module analysis.run module apply_event_filter() (in module mlreco.main_funcs) argmax_nb() (in module mlreco.utils.numba) argmin_nb() (in module mlreco.utils.numba) ARI() (in module mlreco.utils.metrics) as_dict() (mlreco.utils.utils.ForwardData method) ASPP (class in mlreco.models.layers.common.blocks) AtrousIIBlock (class in mlreco.models.layers.common.blocks) attach_prefix() (in module analysis.algorithms.utils) Attention (class in mlreco.models.layers.cluster_cnn.embeddings) AttentionMask (class in mlreco.models.layers.common.ppnplus)

B

backbone_construct() (in module mlreco.models.layers.cluster_cnn.factories) backbone_dict() (in module mlreco.models.layers.cluster_cnn.factories) backward() (mlreco.trainval.trainval method) BatchNorm (class in mlreco.models.layers.gnn.normalizations) bayes_segnet_mcdropout() (in module mlreco.post_processing.metrics.bayes_segnet_mcdropout) BayesianParticleClassifier (class in mlreco.models.singlep) BayesianUResNet (class in mlreco.models.bayes_uresnet) bc_distance() (in module mlreco.models.layers.gnn.losses.node_grouping) BD() (in module mlreco.utils.metrics) BerHuLoss (class in mlreco.models.layers.gnn.losses.node_kinematics)

bhattacharyya_coeff_matrix() (in module mlreco.models.layers.cluster_cnn.losses.misc) bhattacharyya_distance_matrix() (in module mlreco.models.layers.cluster_cnn.losses.misc) bias() (mlreco.models.layers.common.normalizations.MinkowskiAdaIN property) bilinear() (mlreco.models.bayes_uresnet.DUQUResNet method) (mlreco.models.singlep.DUQParticleClassifier method) BilinearKernel (class in mlreco.models.layers.cluster_cnn.gs_kernels) BilinearNNKernel (class in mlreco.models.layers.cluster_cnn.gs_kernels) binary_xloss() (in module mlreco.models.layers.cluster_cnn.losses.lovasz) BinaryCELogDiceLoss (class in mlreco.models.layers.cluster_cnn.losses.misc) BinaryLogDiceLoss (class in mlreco.models.layers.cluster_cnn.losses.misc) BootstrapBatchSampler (class in mlreco.iotools.samplers)

C

calc_gradient_penalty() (mlreco.models.bayes_uresnet.DUQSegmentationLoss static method) (mlreco.models.singlep.MultiLabelCrossEntropy static method) CascadeDilationBlock (class in mlreco.models.layers.common.blocks) cdist_nb() (in module mlreco.utils.numba) CELovaszLoss (class in mlreco.models.layers.cluster_cnn.losses.spatial_embeddings) cfg (mlreco.main_funcs.Handlers attribute) ChunkCSVData (class in mlreco.utils.utils) class_mask() (mlreco.models.spice.SPICELoss method) clean_sparse_data() (in module mlreco.iotools.parsers.clean_data) close() (mlreco.utils.utils.CSVData method) ClustCNNMinkEdgeEncoder (class in mlreco.models.layers.gnn.encoders.cnn) ClustCNNMinkNodeEncoder (class in mlreco.models.layers.gnn.encoders.cnn) cluster_cnn_metrics() (in module mlreco.post_processing.metrics.cluster_cnn_metrics) cluster_gnn_metrics() (in module mlreco.post_processing.metrics.cluster_gnn_metrics) cluster_model_construct() (in module mlreco.models.layers.cluster_cnn.factories) cluster_model_dict() (in module mlreco.models.layers.cluster_cnn.factories) ClusterPool (class in mlreco.models.layers.common.cluster_pool) ClustGeoEdgeEncoder (class in mlreco.models.layers.gnn.encoders.geometric) ClustGeoNodeEncoder (class in mlreco.models.layers.gnn.encoders.geometric) ClustMixEdgeEncoder (class in mlreco.models.layers.gnn.encoders.mixed) ClustMixNodeEncoder (class in mlreco.models.layers.gnn.encoders.mixed) CollateDense() (in module mlreco.iotools.collates) CollateSparse() (in module mlreco.iotools.collates) combine() (mlreco.models.layers.cluster_cnn.losses.single_layers.DiscriminativeLoss method) combine_multiclass() (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.GraphSPICEEmbeddingLoss method) (mlreco.models.layers.cluster_cnn.losses.radius_nnloss.DensityBasedNNLoss method) (mlreco.models.layers.cluster_cnn.losses.single_layers.DiscriminativeLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterBC method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterLoss2 method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast.SPICEInterLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast.SPICELoss method)

compute_edge_weight() (in module mlreco.models.layers.cluster_cnn.losses.gs_embeddings) (mlreco.models.layers.cluster_cnn.gs_kernels.DefaultKernel method) (mlreco.models.layers.cluster_cnn.gs_kernels.MixedKernel method) compute_heuristic_accuracy() (mlreco.models.layers.cluster_cnn.losses.single_layers.DiscriminativeLoss method) compute_loss_single_graph() (mlreco.models.layers.cluster_cnn.losses.misc.VectorEstimationLoss method) compute_metrics() (in module mlreco.post_processing.analysis.muon_residual_range) compute_momentum() (mlreco.models.layers.gnn.losses.node_kinematics.NodeEvidentialKinematicsLoss method) compute_particle_direction() (in module analysis.algorithms.calorimetry) compute_rescaled_charge() (in module mlreco.utils.deghosting) compute_residual_range() (in module mlreco.post_processing.analysis.muon_residual_range) compute_sum_deposited() (in module analysis.algorithms.calorimetry) compute_track_length() (in module analysis.algorithms.calorimetry) compute_type() (mlreco.models.layers.gnn.losses.node_kinematics.NodeEvidentialKinematicsLoss method) (mlreco.models.layers.gnn.losses.node_kinematics.NodeTransformerLoss method) compute_vertex() (mlreco.models.layers.gnn.losses.node_kinematics.NodeEvidentialKinematicsLoss method) (mlreco.models.layers.gnn.losses.node_kinematics.NodeTransformerLoss method) compute_weight_coordinate_space() (mlreco.models.layers.cluster_cnn.gs_kernels.MixedKernel method) construct() (in module mlreco.models.factories) contains() (in module mlreco.utils.ppn) contingency_table() (in module mlreco.utils.metrics) ConvolutionBlock (class in mlreco.models.layers.common.blocks) cosmic_discriminator_metrics() (in module mlreco.post_processing.metrics.cosmic_discriminator_metrics) count_primary_particles() (in module analysis.algorithms.utils) covariance_loss() (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.GraphSPICEEmbeddingLoss method) create() (mlreco.iotools.datasets.LArCVDataset static method) (mlreco.iotools.samplers.BootstrapBatchSampler static method) (mlreco.iotools.samplers.RandomSequenceSampler static method) (mlreco.iotools.samplers.SequentialBatchSampler static method) csv_logger (mlreco.main_funcs.Handlers attribute) CSVData (class in mlreco.utils.utils) cycle() (in module mlreco.main_funcs)

D

data_io (mlreco.main_funcs.Handlers attribute) data_io_iter (mlreco.main_funcs.Handlers attribute) data_keys() (mlreco.iotools.datasets.LArCVDataset method) DataParallel (class in mlreco.utils.data_parallel) dataset_factory() (in module mlreco.iotools.factories) dbscan_groups() (in module mlreco.utils.dbscan) dbscan_points() (in module mlreco.utils.dbscan) dbscan_types() (in module mlreco.utils.dbscan) DBSCANFragmenter (class in mlreco.models.layers.common.dbscan) decoder() (mlreco.models.layers.common.fpn.FPN method) (mlreco.models.layers.common.mobilenet.MobileNetV3 method) (mlreco.models.layers.common.senet.SENet method) (mlreco.models.layers.common.uresnet_layers.UResNetDecoder method) (mlreco.models.layers.common.uresnext.UResNeXt method) DeepGraphTransformer (class in mlreco.models.layers.gnn.message_passing.dgtransformer)

DeepVertexNet (class in mlreco.models.layers.common.momentum) default_gnn_metrics() (in module mlreco.post_processing.metrics.evidential_gnn) DefaultKernel (class in mlreco.models.layers.cluster_cnn.gs_kernels) deghost_labels_and_predictions() (in module mlreco.utils.deghosting) deghosting_metrics() (in module mlreco.post_processing.metrics.deghosting_metrics) dense_coordinates() (in module mlreco.models.layers.common.blocks) DenseResBlock (class in mlreco.models.layers.common.blocks) DensityBasedNNLoss (class in mlreco.models.layers.cluster_cnn.losses.radius_nnloss) DiscriminativeLoss (class in mlreco.models.layers.cluster_cnn.losses.single_layers) DropoutBlock (class in mlreco.models.layers.common.blocks) dualplot() (in module mlreco.visualization.plotly_layouts) duq_metrics() (in module mlreco.post_processing.metrics.duq_metrics) DUQParticleClassifier (class in mlreco.models.singlep) DUQSegmentationLoss (class in mlreco.models.bayes_uresnet) DUQUResNet (class in mlreco.models.bayes_uresnet)

E

EConvModel (class in mlreco.models.layers.gnn.message_passing.econv) edge_encoder_construct() (in module mlreco.models.layers.gnn.factories) edge_encoder_dict() (in module mlreco.models.layers.gnn.factories) edge_loss_construct() (in module mlreco.models.layers.gnn.factories) edge_loss_dict() (in module mlreco.models.layers.gnn.factories) EdgeChannelLoss (class in mlreco.models.layers.gnn.losses.edge_channel) EdgeLayer (class in mlreco.models.layers.gnn.message_passing.agnnconv) (class in mlreco.models.layers.gnn.message_passing.econv) (class in mlreco.models.layers.gnn.message_passing.gatconv) (class in mlreco.models.layers.gnn.message_passing.meta) (class in mlreco.models.layers.gnn.message_passing.nnconv) (class in mlreco.models.layers.gnn.message_passing.nnconv_elu) (class in mlreco.models.layers.gnn.message_passing.nnconv_old) efficiency() (in module mlreco.utils.metrics) embed() (mlreco.models.bayes_uresnet.DUQUResNet method) (mlreco.models.singlep.DUQParticleClassifier method) encoder() (mlreco.models.layers.common.fpn.FPN method) (mlreco.models.layers.common.mobilenet.MB3Encoder method) (mlreco.models.layers.common.mobilenet.MobileNetV3 method) (mlreco.models.layers.common.senet.SENet method) (mlreco.models.layers.common.uresnet_layers.UResNetEncoder method) (mlreco.models.layers.common.uresnext.UResNeXt method)

eps (mlreco.models.layers.gnn.normalizations.BatchNorm attribute) (mlreco.models.layers.gnn.normalizations.InstanceNorm attribute) evaluate() (in module analysis.decorator) evidential_forward() (mlreco.models.bayes_uresnet.BayesianUResNet method) (mlreco.models.singlep.BayesianParticleClassifier method) evidential_gnn_metrics() (in module mlreco.post_processing.metrics.evidential_gnn) evidential_metrics() (in module mlreco.post_processing.metrics.evidential_metrics) evidential_segnet_metrics() (in module mlreco.post_processing.metrics.evidential_segnet) EvidentialLearningLoss (class in mlreco.models.singlep) EvidentialMomentumNet (class in mlreco.models.layers.common.momentum) EvidentialParticleClassifier (class in mlreco.models.singlep) ExpandAs (class in mlreco.models.layers.cluster_cnn.embeddings) (class in mlreco.models.layers.common.ppnplus) expansion (mlreco.models.layers.common.blocks.ResNetBlock attribute) (mlreco.models.layers.common.blocks.SEResNetBlock attribute) extent() (in module mlreco.post_processing.common)

F

feature_embedding_loss() (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.GraphSPICEEmbeddingLoss method) filter_class() (mlreco.models.graph_spice.GraphSPICELoss method) (mlreco.models.graph_spice.MinkGraphSPICE method) (mlreco.models.layers.cluster_cnn.embeddings.SPICE method) filter_duplicate_voxels() (in module mlreco.utils.groups) filter_duplicate_voxels_ref() (in module mlreco.utils.groups) filter_group_data() (in module mlreco.utils.groups) filter_nonimg_voxels() (in module mlreco.utils.groups) find_closest_points_of_approach() (in module mlreco.utils.vertex) find_cluster_means() (in module mlreco.models.layers.cluster_cnn.losses.misc) (mlreco.models.layers.cluster_cnn.losses.single_layers.DiscriminativeLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast.SPICELoss method) find_cosmic_angle() (in module analysis.algorithms.selections.michel_electrons) find_edges() (in module mlreco.post_processing.analysis.michel_reconstruction_2d) find_true_cosmic_angle() (in module analysis.algorithms.selections.michel_electrons) fit_predict() (analysis.classes.ui.FullChainPredictor method) fit_predict_labels() (analysis.classes.ui.FullChainPredictor method) flatten_binary_scores() (in module mlreco.models.layers.cluster_cnn.losses.lovasz) flatten_probas() (in module mlreco.models.layers.cluster_cnn.losses.lovasz) flush() (mlreco.utils.utils.CSVData method) FocalLoss (class in mlreco.models.layers.cluster_cnn.losses.misc) forward() (mlreco.models.bayes_uresnet.BayesianUResNet method) (mlreco.models.bayes_uresnet.DUQSegmentationLoss method) (mlreco.models.bayes_uresnet.DUQUResNet method) (mlreco.models.bayes_uresnet.SegmentationLoss method) (mlreco.models.graph_spice.GraphSPICELoss method) (mlreco.models.graph_spice.MinkGraphSPICE method) (mlreco.models.grappa.GNN method) (mlreco.models.grappa.GNNLoss method) (mlreco.models.layers.cluster_cnn.embeddings.Attention method) (mlreco.models.layers.cluster_cnn.embeddings.ExpandAs method) (mlreco.models.layers.cluster_cnn.embeddings.SPICE method) (mlreco.models.layers.cluster_cnn.graph_spice_embedder.GraphSPICEEmbedder method) (mlreco.models.layers.cluster_cnn.gs_kernels.BilinearKernel method) (mlreco.models.layers.cluster_cnn.gs_kernels.BilinearNNKernel method) (mlreco.models.layers.cluster_cnn.gs_kernels.DefaultKernel method) (mlreco.models.layers.cluster_cnn.gs_kernels.MixedKernel method) (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.GraphSPICEEmbeddingLoss method) (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.NodeEdgeHybridLoss method) (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.WeightedEdgeLoss method) (mlreco.models.layers.cluster_cnn.losses.lovasz.StableBCELoss method) (mlreco.models.layers.cluster_cnn.losses.misc.BinaryCELogDiceLoss method) (mlreco.models.layers.cluster_cnn.losses.misc.BinaryLogDiceLoss method) (mlreco.models.layers.cluster_cnn.losses.misc.FocalLoss method) (mlreco.models.layers.cluster_cnn.losses.misc.IoUScore method) (mlreco.models.layers.cluster_cnn.losses.misc.LovaszHingeLoss method) (mlreco.models.layers.cluster_cnn.losses.misc.LovaszSoftmaxWithLogitsLoss method) (mlreco.models.layers.cluster_cnn.losses.misc.MincutLoss method) (mlreco.models.layers.cluster_cnn.losses.misc.VectorEstimationLoss method) (mlreco.models.layers.cluster_cnn.losses.misc.WeightedFocalLoss method) (mlreco.models.layers.cluster_cnn.losses.radius_nnloss.DensityBasedNNLoss method) (mlreco.models.layers.cluster_cnn.losses.single_layers.DiscriminativeLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast.SPICELoss method) (mlreco.models.layers.common.blocks.ASPP method) (mlreco.models.layers.common.blocks.AtrousIIBlock method) (mlreco.models.layers.common.blocks.CascadeDilationBlock method) (mlreco.models.layers.common.blocks.ConvolutionBlock method) (mlreco.models.layers.common.blocks.DenseResBlock method) (mlreco.models.layers.common.blocks.DropoutBlock method) (mlreco.models.layers.common.blocks.Identity method) (mlreco.models.layers.common.blocks.MBConv method) (mlreco.models.layers.common.blocks.MBResConv method) (mlreco.models.layers.common.blocks.MBResConvSE method) (mlreco.models.layers.common.blocks.ResNetBlock method) (mlreco.models.layers.common.blocks.ResNeXtBlock method) (mlreco.models.layers.common.blocks.SEBlock method) (mlreco.models.layers.common.blocks.SEResNetBlock method) (mlreco.models.layers.common.blocks.SparseToDense method) (mlreco.models.layers.common.blocks.SPP method) (mlreco.models.layers.common.cluster_pool.ClusterPool method) (mlreco.models.layers.common.cnn_encoder.SparseResidualEncoder method) (mlreco.models.layers.common.dbscan.DBSCANFragmenter method) (mlreco.models.layers.common.extract_feature_map.MinkGhostMask method) (mlreco.models.layers.common.fpn.FPN method) (mlreco.models.layers.common.gnn_full_chain.FullChainGNN method) (mlreco.models.layers.common.gnn_full_chain.FullChainLoss method) (mlreco.models.layers.common.mobilenet.MB3Encoder method) (mlreco.models.layers.common.mobilenet.MobileNetV3 method) (mlreco.models.layers.common.momentum.DeepVertexNet method) (mlreco.models.layers.common.momentum.EvidentialMomentumNet method) (mlreco.models.layers.common.momentum.MomentumNet method) (mlreco.models.layers.common.momentum.VertexNet method) (mlreco.models.layers.common.nonlinearities.MinkowskiMish method) (mlreco.models.layers.common.normalizations.MinkowskiAdaIN method) (mlreco.models.layers.common.normalizations.MinkowskiPixelNorm method) (mlreco.models.layers.common.ppnplus.AttentionMask method) (mlreco.models.layers.common.ppnplus.ExpandAs method) (mlreco.models.layers.common.ppnplus.MergeConcat method) (mlreco.models.layers.common.ppnplus.PPN method) (mlreco.models.layers.common.ppnplus.PPNLonelyLoss method) (mlreco.models.layers.common.senet.SENet method) (mlreco.models.layers.common.sparse_generator.SparseGenerator method) (mlreco.models.layers.common.sparse_generator.SparseGeneratorSimple method) (mlreco.models.layers.common.uresnet_layers.UResNet method) (mlreco.models.layers.common.uresnet_layers.UResNetDecoder method) (mlreco.models.layers.common.uresnet_layers.UResNetEncoder method) (mlreco.models.layers.common.uresnext.UResNeXt method) (mlreco.models.layers.gnn.encoders.cnn.ClustCNNMinkEdgeEncoder method) (mlreco.models.layers.gnn.encoders.cnn.ClustCNNMinkNodeEncoder method) (mlreco.models.layers.gnn.encoders.geometric.ClustGeoEdgeEncoder method) (mlreco.models.layers.gnn.encoders.geometric.ClustGeoNodeEncoder method) (mlreco.models.layers.gnn.encoders.mixed.ClustMixEdgeEncoder method) (mlreco.models.layers.gnn.encoders.mixed.ClustMixNodeEncoder method) (mlreco.models.layers.gnn.losses.edge_channel.EdgeChannelLoss method) (mlreco.models.layers.gnn.losses.node_grouping.GNNGroupingLoss method) (mlreco.models.layers.gnn.losses.node_kinematics.BerHuLoss method) (mlreco.models.layers.gnn.losses.node_kinematics.LogRMSE method) (mlreco.models.layers.gnn.losses.node_kinematics.NodeEvidentialKinematicsLoss method) (mlreco.models.layers.gnn.losses.node_kinematics.NodeKinematicsLoss method) (mlreco.models.layers.gnn.losses.node_primary.NodePrimaryLoss method) (mlreco.models.layers.gnn.losses.node_type.NodeTypeLoss method) (mlreco.models.layers.gnn.message_passing.agnnconv.AGNNConvModel method) (mlreco.models.layers.gnn.message_passing.agnnconv.EdgeLayer method) (mlreco.models.layers.gnn.message_passing.agnnconv.GlobalModel method) (mlreco.models.layers.gnn.message_passing.agnnconv.NodeLayer method) (mlreco.models.layers.gnn.message_passing.econv.EConvModel method) (mlreco.models.layers.gnn.message_passing.econv.EdgeLayer method) (mlreco.models.layers.gnn.message_passing.econv.GlobalModel method) (mlreco.models.layers.gnn.message_passing.econv.NodeLayer method) (mlreco.models.layers.gnn.message_passing.gatconv.EdgeLayer method) (mlreco.models.layers.gnn.message_passing.gatconv.GATConvModel method) (mlreco.models.layers.gnn.message_passing.gatconv.GlobalModel method) (mlreco.models.layers.gnn.message_passing.gatconv.NodeLayer method) (mlreco.models.layers.gnn.message_passing.meta.EdgeLayer method) (mlreco.models.layers.gnn.message_passing.meta.GlobalModel method) (mlreco.models.layers.gnn.message_passing.meta.MetaLayerModel method) (mlreco.models.layers.gnn.message_passing.meta.NodeLayer method) (mlreco.models.layers.gnn.message_passing.nnconv.EdgeLayer method) (mlreco.models.layers.gnn.message_passing.nnconv.GlobalModel method) (mlreco.models.layers.gnn.message_passing.nnconv.NNConvModel method) (mlreco.models.layers.gnn.message_passing.nnconv.NodeLayer method) (mlreco.models.layers.gnn.message_passing.nnconv_elu.EdgeLayer method) (mlreco.models.layers.gnn.message_passing.nnconv_elu.GlobalModel method) (mlreco.models.layers.gnn.message_passing.nnconv_elu.NNConvModel method) (mlreco.models.layers.gnn.message_passing.nnconv_elu.NodeLayer method) (mlreco.models.layers.gnn.message_passing.nnconv_old.EdgeLayer method) (mlreco.models.layers.gnn.message_passing.nnconv_old.GlobalModel method) (mlreco.models.layers.gnn.message_passing.nnconv_old.NNConvModel method) (mlreco.models.layers.gnn.message_passing.nnconv_old.NodeLayer method) (mlreco.models.layers.gnn.normalizations.BatchNorm method) (mlreco.models.layers.gnn.normalizations.InstanceNorm method) (mlreco.models.singlep.BayesianParticleClassifier method) (mlreco.models.singlep.DUQParticleClassifier method) (mlreco.models.singlep.EvidentialLearningLoss method) (mlreco.models.singlep.EvidentialParticleClassifier method) (mlreco.models.singlep.MultiLabelCrossEntropy method) (mlreco.models.singlep.ParticleImageClassifier method) (mlreco.models.singlep.ParticleTypeLoss method) (mlreco.models.spice.SPICELoss method) (mlreco.models.uresnet.SegmentationLoss method) (mlreco.models.uresnet.UResNet_Chain method) (mlreco.models.uresnet_ppn_chain.UResNetPPN method) (mlreco.models.uresnet_ppn_chain.UResNetPPNLoss method) (mlreco.trainval.trainval method)

ForwardData (class in mlreco.utils.utils) FPN (class in mlreco.models.layers.common.fpn) freeze_weights() (mlreco.trainval.trainval method) full_chain_cnn() (mlreco.models.full_chain.FullChain method) full_chain_gnn() (mlreco.models.layers.common.gnn_full_chain.FullChainGNN method) FullChain (class in mlreco.models.full_chain) FullChainEvaluator (class in analysis.classes.ui) FullChainGNN (class in mlreco.models.layers.common.gnn_full_chain) FullChainLoss (class in mlreco.models.full_chain) (class in mlreco.models.layers.common.gnn_full_chain) FullChainPredictor (class in analysis.classes.ui)

G

GATConvModel (class in mlreco.models.layers.gnn.message_passing.gatconv) gather() (mlreco.utils.data_parallel.DataParallel method) gauss() (in module mlreco.models.layers.gnn.cluster) get_all_fragments() (mlreco.models.layers.common.gnn_full_chain.FullChainGNN method) get_all_particles() (mlreco.models.layers.common.gnn_full_chain.FullChainGNN method) get_batch_indices() (mlreco.models.layers.common.sparse_generator.SparseGenerator method) (mlreco.models.layers.common.sparse_generator.SparseGeneratorSimple method) get_bounding_box() (in module analysis.algorithms.selections.michel_electrons) get_clusts() (mlreco.models.layers.common.dbscan.DBSCANFragmenter method) get_coords() (in module mlreco.post_processing.store.store_input) get_data_minibatched() (mlreco.trainval.trainval method) get_embeddings() (mlreco.models.layers.cluster_cnn.graph_spice_embedder.GraphSPICEEmbedder method) get_event_list() (mlreco.iotools.datasets.LArCVDataset static method) get_extra_gnn_features() (mlreco.models.full_chain.FullChain static method) get_flash_matches() (analysis.classes.ui.FullChainPredictor method) get_fragments() (analysis.classes.ui.FullChainPredictor method) (in module mlreco.post_processing.analysis.muon_residual_range) get_graphspice_logits() (in module mlreco.models.layers.cluster_cnn.losses.misc) get_group_primary_id() (in module mlreco.utils.groups) get_group_types() (in module mlreco.utils.groups) get_interaction_id() (in module mlreco.utils.groups) get_interactions() (analysis.classes.ui.FullChainPredictor method) get_nu_id() (in module mlreco.utils.groups) get_object_color() (in module mlreco.visualization.gnn) get_particle_id() (in module mlreco.utils.groups) get_particle_properties() (in module analysis.algorithms.utils) get_particles() (analysis.classes.ui.FullChainPredictor method) get_per_class_probabilities() (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.CELovaszLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELoss2 method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELossBivariate method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskFocalLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszHingeLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterBC method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterLoss2 method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MultiVariateLovasz method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast.SPICEInterLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast.SPICELoss method)

get_ppn_info() (in module mlreco.utils.ppn) get_ppn_points_per_particles() (in module mlreco.utils.vertex) get_predicted_label() (analysis.classes.ui.FullChainEvaluator method) get_probs() (in module mlreco.models.layers.cluster_cnn.losses.misc) get_shower_primary_id() (in module mlreco.utils.groups) get_shower_startpoint() (in module analysis.algorithms.point_matching) get_target() (mlreco.models.layers.common.sparse_generator.SparseGenerator method) (mlreco.models.layers.common.sparse_generator.SparseGeneratorSimple method) get_track_endpoints() (in module analysis.algorithms.point_matching) get_track_endpoints_centroid() (in module analysis.algorithms.point_matching) get_track_endpoints_geo() (in module mlreco.utils.ppn) get_true_fragments() (analysis.classes.ui.FullChainEvaluator method) get_true_interactions() (analysis.classes.ui.FullChainEvaluator method) get_true_label() (analysis.classes.ui.FullChainEvaluator method) get_true_particles() (analysis.classes.ui.FullChainEvaluator method) get_true_vertices() (analysis.classes.ui.FullChainEvaluator method) get_vertex() (in module mlreco.utils.vertex) global_efficiency() (in module mlreco.utils.metrics) global_purity() (in module mlreco.utils.metrics) GlobalModel (class in mlreco.models.layers.gnn.message_passing.agnnconv) (class in mlreco.models.layers.gnn.message_passing.econv) (class in mlreco.models.layers.gnn.message_passing.gatconv) (class in mlreco.models.layers.gnn.message_passing.meta) (class in mlreco.models.layers.gnn.message_passing.nnconv) (class in mlreco.models.layers.gnn.message_passing.nnconv_elu) (class in mlreco.models.layers.gnn.message_passing.nnconv_old) GNN (class in mlreco.models.grappa) gnn_model_construct() (in module mlreco.models.layers.gnn.factories) gnn_model_dict() (in module mlreco.models.layers.gnn.factories) GNNGroupingLoss (class in mlreco.models.layers.gnn.losses.node_grouping) GNNLoss (class in mlreco.models.grappa) graph_spice_metrics() (in module mlreco.post_processing.metrics.graph_spice_metrics) graph_spice_metrics_loop_threshold() (in module mlreco.post_processing.metrics.graph_spice_metrics) GraphSPICEEmbedder (class in mlreco.models.layers.cluster_cnn.graph_spice_embedder) GraphSPICEEmbeddingLoss (class in mlreco.models.layers.cluster_cnn.losses.gs_embeddings) GraphSPICELoss (class in mlreco.models.graph_spice) group_particles_to_interactions_fn() (in module analysis.classes.particle) group_points() (in module mlreco.utils.ppn) gs_kernel_construct() (in module mlreco.models.layers.cluster_cnn.factories) gs_kernel_dict() (in module mlreco.models.layers.cluster_cnn.factories)

H

Handlers (class in mlreco.main_funcs)

I

Identity (class in mlreco.models.layers.common.blocks) inference() (in module mlreco.main_funcs) inference_loop() (in module mlreco.main_funcs) initialize() (mlreco.trainval.trainval method) initialize_calibrator() (mlreco.trainval.trainval method) INPUT_SCHEMA (mlreco.models.uresnet.SegmentationLoss attribute) (mlreco.models.uresnet.UResNet_Chain attribute) instance_clustering() (in module mlreco.post_processing.analysis.instance_clustering) InstanceNorm (class in mlreco.models.layers.gnn.normalizations) inter_cluster_loss() (in module mlreco.models.layers.cluster_cnn.losses.misc) (mlreco.models.layers.cluster_cnn.losses.single_layers.DiscriminativeLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterBC method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast.SPICEInterLoss method)

intra_cluster_loss() (in module mlreco.models.layers.cluster_cnn.losses.misc) (mlreco.models.layers.cluster_cnn.losses.single_layers.DiscriminativeLoss method) iou() (in module mlreco.models.layers.cluster_cnn.losses.lovasz) iou_batch() (in module mlreco.models.layers.cluster_cnn.losses.misc) iou_binary() (in module mlreco.models.layers.cluster_cnn.losses.lovasz) IoUScore (class in mlreco.models.layers.cluster_cnn.losses.misc) is_at_edge() (in module mlreco.post_processing.analysis.michel_reconstruction_2d) is_attached_at_edge() (in module analysis.algorithms.selections.michel_electrons) is_contained() (analysis.classes.ui.FullChainPredictor method) isnan() (in module mlreco.models.layers.cluster_cnn.losses.lovasz) iteration (mlreco.main_funcs.Handlers attribute)

K

keys() (mlreco.main_funcs.Handlers method)

kinematics_metrics() (in module mlreco.post_processing.metrics.kinematics_metrics)

L

LABEL_TO_COLUMN (analysis.classes.ui.FullChainEvaluator attribute) laplace() (in module mlreco.models.layers.gnn.cluster) LArCVDataset (class in mlreco.iotools.datasets) list_concat() (in module mlreco.utils.unwrap) list_data() (mlreco.iotools.datasets.LArCVDataset static method) load_weights() (mlreco.trainval.trainval method) loader_factory() (in module mlreco.iotools.factories) local_cdist() (in module mlreco.utils.utils) log() (in module mlreco.main_funcs)

log_loss_nb() (in module mlreco.utils.numba) logit_fn() (in module mlreco.models.layers.cluster_cnn.losses.misc) LogRMSE (class in mlreco.models.layers.gnn.losses.node_kinematics) lovasz_grad() (in module mlreco.models.layers.cluster_cnn.losses.lovasz) lovasz_hinge() (in module mlreco.models.layers.cluster_cnn.losses.lovasz) lovasz_hinge_flat() (in module mlreco.models.layers.cluster_cnn.losses.lovasz) lovasz_softmax() (in module mlreco.models.layers.cluster_cnn.losses.lovasz) lovasz_softmax_flat() (in module mlreco.models.layers.cluster_cnn.losses.lovasz) LovaszHingeLoss (class in mlreco.models.layers.cluster_cnn.losses.misc) LovaszSoftmaxWithLogitsLoss (class in mlreco.models.layers.cluster_cnn.losses.misc)

M

main() (in module analysis.run) make_directories() (in module mlreco.main_funcs) make_input_forward() (mlreco.trainval.trainval method) margin_smoothing_loss() (in module mlreco.models.layers.cluster_cnn.losses.misc) MaskBCELoss (class in mlreco.models.layers.cluster_cnn.losses.spatial_embeddings) MaskBCELoss2 (class in mlreco.models.layers.cluster_cnn.losses.spatial_embeddings) MaskBCELossBivariate (class in mlreco.models.layers.cluster_cnn.losses.spatial_embeddings) MaskFocalLoss (class in mlreco.models.layers.cluster_cnn.losses.spatial_embeddings) MaskLovaszHingeLoss (class in mlreco.models.layers.cluster_cnn.losses.spatial_embeddings) MaskLovaszInterBC (class in mlreco.models.layers.cluster_cnn.losses.spatial_embeddings) MaskLovaszInterLoss (class in mlreco.models.layers.cluster_cnn.losses.spatial_embeddings) MaskLovaszInterLoss2 (class in mlreco.models.layers.cluster_cnn.losses.spatial_embeddings) match_interactions() (analysis.classes.ui.FullChainEvaluator method) match_interactions_fn() (in module analysis.classes.particle) match_particles() (analysis.classes.ui.FullChainEvaluator method) match_particles_fn() (in module analysis.classes.particle) match_points_to_particles() (in module analysis.algorithms.point_matching) matrix_counts() (in module analysis.classes.particle) matrix_iou() (in module analysis.classes.particle) max_nb() (in module mlreco.utils.numba) MB3Encoder (class in mlreco.models.layers.common.mobilenet) MBConv (class in mlreco.models.layers.common.blocks) MBResConv (class in mlreco.models.layers.common.blocks) MBResConvSE (class in mlreco.models.layers.common.blocks) mc_forward() (mlreco.models.bayes_uresnet.BayesianUResNet method) (mlreco.models.singlep.BayesianParticleClassifier method) mean() (in module mlreco.models.layers.cluster_cnn.losses.lovasz) mean_nb() (in module mlreco.utils.numba) MergeConcat (class in mlreco.models.layers.common.ppnplus) MetaLayerModel (class in mlreco.models.layers.gnn.message_passing.meta) michel_electrons() (in module analysis.algorithms.selections.michel_electrons) michel_reconstruction() (in module mlreco.post_processing.analysis.michel_reconstruction) michel_reconstruction_2d() (in module mlreco.post_processing.analysis.michel_reconstruction_2d) michel_reconstruction_noghost() (in module mlreco.post_processing.analysis.michel_reconstruction_noghost) min_nb() (in module mlreco.utils.numba) MincutLoss (class in mlreco.models.layers.cluster_cnn.losses.misc) MinkGhostMask (class in mlreco.models.layers.common.extract_feature_map) MinkGraphSPICE (class in mlreco.models.graph_spice) MinkowskiAdaIN (class in mlreco.models.layers.common.normalizations) MinkowskiELU (class in mlreco.models.layers.common.nonlinearities) MinkowskiGroupNorm (class in mlreco.models.layers.common.normalizations) MinkowskiLeakyReLU (class in mlreco.models.layers.common.nonlinearities) MinkowskiMish (class in mlreco.models.layers.common.nonlinearities) MinkowskiPixelNorm (class in mlreco.models.layers.common.normalizations) MinkSPICE (class in mlreco.models.spice) MixedKernel (class in mlreco.models.layers.cluster_cnn.gs_kernels) mlreco module mlreco.iotools module mlreco.iotools.collates module mlreco.iotools.datasets module mlreco.iotools.factories module mlreco.iotools.parsers module mlreco.iotools.parsers.clean_data module mlreco.iotools.parsers.cluster module mlreco.iotools.parsers.misc module mlreco.iotools.parsers.particles module mlreco.iotools.parsers.sparse module mlreco.iotools.samplers module mlreco.main_funcs module mlreco.models module mlreco.models.bayes_uresnet module mlreco.models.factories module mlreco.models.full_chain module mlreco.models.graph_spice module mlreco.models.grappa module mlreco.models.layers module mlreco.models.layers.cluster_cnn module mlreco.models.layers.cluster_cnn.embeddings module mlreco.models.layers.cluster_cnn.factories module mlreco.models.layers.cluster_cnn.graph_spice_embedder module mlreco.models.layers.cluster_cnn.gs_kernels module mlreco.models.layers.cluster_cnn.losses module mlreco.models.layers.cluster_cnn.losses.gs_embeddings module mlreco.models.layers.cluster_cnn.losses.lovasz module mlreco.models.layers.cluster_cnn.losses.misc module mlreco.models.layers.cluster_cnn.losses.radius_nnloss module mlreco.models.layers.cluster_cnn.losses.single_layers module mlreco.models.layers.cluster_cnn.losses.spatial_embeddings module mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast module mlreco.models.layers.common module mlreco.models.layers.common.activation_normalization_factories module mlreco.models.layers.common.blocks module mlreco.models.layers.common.cluster_pool module mlreco.models.layers.common.cnn_encoder module mlreco.models.layers.common.configuration module mlreco.models.layers.common.dbscan module mlreco.models.layers.common.extract_feature_map module mlreco.models.layers.common.fpn module mlreco.models.layers.common.gnn_full_chain module mlreco.models.layers.common.mobilenet module mlreco.models.layers.common.momentum module mlreco.models.layers.common.nonlinearities module mlreco.models.layers.common.normalizations module mlreco.models.layers.common.ppnplus module mlreco.models.layers.common.senet module mlreco.models.layers.common.sparse_generator module mlreco.models.layers.common.uresnet_layers module mlreco.models.layers.common.uresnext module mlreco.models.layers.gnn module mlreco.models.layers.gnn.cluster module mlreco.models.layers.gnn.encoders module mlreco.models.layers.gnn.encoders.cnn module mlreco.models.layers.gnn.encoders.geometric module mlreco.models.layers.gnn.encoders.mixed module mlreco.models.layers.gnn.factories module mlreco.models.layers.gnn.losses module mlreco.models.layers.gnn.losses.edge_channel module mlreco.models.layers.gnn.losses.node_grouping module mlreco.models.layers.gnn.losses.node_kinematics module mlreco.models.layers.gnn.losses.node_primary module mlreco.models.layers.gnn.losses.node_type module mlreco.models.layers.gnn.message_passing module mlreco.models.layers.gnn.message_passing.agnnconv module mlreco.models.layers.gnn.message_passing.dgtransformer module mlreco.models.layers.gnn.message_passing.econv module mlreco.models.layers.gnn.message_passing.gatconv module mlreco.models.layers.gnn.message_passing.meta module mlreco.models.layers.gnn.message_passing.nnconv module mlreco.models.layers.gnn.message_passing.nnconv_elu module mlreco.models.layers.gnn.message_passing.nnconv_old module mlreco.models.layers.gnn.normalizations module mlreco.models.singlep module mlreco.models.spice module mlreco.models.uresnet module mlreco.models.uresnet_ppn_chain module mlreco.post_processing module mlreco.post_processing.analysis module mlreco.post_processing.analysis.instance_clustering module mlreco.post_processing.analysis.michel_reconstruction module mlreco.post_processing.analysis.michel_reconstruction_2d module mlreco.post_processing.analysis.michel_reconstruction_noghost module mlreco.post_processing.analysis.muon_residual_range module mlreco.post_processing.analysis.nue_selection module mlreco.post_processing.analysis.stopping_muons module mlreco.post_processing.analysis.through_muons module mlreco.post_processing.analysis.track_clustering module mlreco.post_processing.common module mlreco.post_processing.decorator module mlreco.post_processing.metrics module mlreco.post_processing.metrics.bayes_segnet_mcdropout module mlreco.post_processing.metrics.cluster_cnn_metrics module mlreco.post_processing.metrics.cluster_gnn_metrics module mlreco.post_processing.metrics.cosmic_discriminator_metrics module mlreco.post_processing.metrics.deghosting_metrics module mlreco.post_processing.metrics.duq_metrics module mlreco.post_processing.metrics.evidential_gnn module mlreco.post_processing.metrics.evidential_metrics module mlreco.post_processing.metrics.evidential_segnet module

mlreco.post_processing.metrics.graph_spice_metrics module mlreco.post_processing.metrics.kinematics_metrics module mlreco.post_processing.metrics.pid_metrics module mlreco.post_processing.metrics.ppn_metrics module mlreco.post_processing.metrics.ppn_simple module mlreco.post_processing.metrics.single_particle module mlreco.post_processing.metrics.singlep_mcdropout module mlreco.post_processing.metrics.uresnet_metrics module mlreco.post_processing.metrics.vertex_metrics module mlreco.post_processing.store module mlreco.post_processing.store.store_input module mlreco.post_processing.store.store_output module mlreco.post_processing.store.store_uresnet module mlreco.post_processing.store.store_uresnet_ppn module mlreco.trainval module mlreco.utils module mlreco.utils.adabound module mlreco.utils.data_parallel module mlreco.utils.dbscan module mlreco.utils.deghosting module mlreco.utils.groups module mlreco.utils.metrics module mlreco.utils.numba module mlreco.utils.ppn module mlreco.utils.track_clustering module mlreco.utils.unwrap module mlreco.utils.utils module mlreco.utils.vertex module mlreco.visualization module mlreco.visualization.cubes module mlreco.visualization.gnn module mlreco.visualization.plotly_layouts module mlreco.visualization.points module mlreco.visualization.voxels module MobileNetV3 (class in mlreco.models.layers.common.mobilenet) model_dict() (in module mlreco.models.factories) modified_ARI() (in module mlreco.post_processing.metrics.graph_spice_metrics) modified_efficiency() (in module mlreco.post_processing.metrics.graph_spice_metrics) modified_purity() (in module mlreco.post_processing.metrics.graph_spice_metrics) module analysis analysis.algorithms analysis.algorithms.calorimetry analysis.algorithms.point_matching analysis.algorithms.selections analysis.algorithms.selections.michel_electrons analysis.algorithms.selections.stopping_muons analysis.algorithms.selections.through_going_muons analysis.algorithms.utils analysis.classes analysis.classes.particle analysis.classes.ui analysis.decorator analysis.run mlreco mlreco.iotools mlreco.iotools.collates mlreco.iotools.datasets mlreco.iotools.factories mlreco.iotools.parsers mlreco.iotools.parsers.clean_data mlreco.iotools.parsers.cluster mlreco.iotools.parsers.misc mlreco.iotools.parsers.particles mlreco.iotools.parsers.sparse mlreco.iotools.samplers mlreco.main_funcs mlreco.models mlreco.models.bayes_uresnet mlreco.models.factories mlreco.models.full_chain mlreco.models.graph_spice mlreco.models.grappa mlreco.models.layers mlreco.models.layers.cluster_cnn mlreco.models.layers.cluster_cnn.embeddings mlreco.models.layers.cluster_cnn.factories mlreco.models.layers.cluster_cnn.graph_spice_embedder mlreco.models.layers.cluster_cnn.gs_kernels mlreco.models.layers.cluster_cnn.losses mlreco.models.layers.cluster_cnn.losses.gs_embeddings mlreco.models.layers.cluster_cnn.losses.lovasz mlreco.models.layers.cluster_cnn.losses.misc mlreco.models.layers.cluster_cnn.losses.radius_nnloss mlreco.models.layers.cluster_cnn.losses.single_layers mlreco.models.layers.cluster_cnn.losses.spatial_embeddings mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast mlreco.models.layers.common mlreco.models.layers.common.activation_normalization_factories mlreco.models.layers.common.blocks mlreco.models.layers.common.cluster_pool mlreco.models.layers.common.cnn_encoder mlreco.models.layers.common.configuration mlreco.models.layers.common.dbscan mlreco.models.layers.common.extract_feature_map mlreco.models.layers.common.fpn mlreco.models.layers.common.gnn_full_chain mlreco.models.layers.common.mobilenet mlreco.models.layers.common.momentum mlreco.models.layers.common.nonlinearities mlreco.models.layers.common.normalizations mlreco.models.layers.common.ppnplus mlreco.models.layers.common.senet mlreco.models.layers.common.sparse_generator mlreco.models.layers.common.uresnet_layers mlreco.models.layers.common.uresnext mlreco.models.layers.gnn mlreco.models.layers.gnn.cluster mlreco.models.layers.gnn.encoders mlreco.models.layers.gnn.encoders.cnn mlreco.models.layers.gnn.encoders.geometric mlreco.models.layers.gnn.encoders.mixed mlreco.models.layers.gnn.factories mlreco.models.layers.gnn.losses mlreco.models.layers.gnn.losses.edge_channel mlreco.models.layers.gnn.losses.node_grouping mlreco.models.layers.gnn.losses.node_kinematics mlreco.models.layers.gnn.losses.node_primary mlreco.models.layers.gnn.losses.node_type mlreco.models.layers.gnn.message_passing mlreco.models.layers.gnn.message_passing.agnnconv mlreco.models.layers.gnn.message_passing.dgtransformer mlreco.models.layers.gnn.message_passing.econv mlreco.models.layers.gnn.message_passing.gatconv mlreco.models.layers.gnn.message_passing.meta mlreco.models.layers.gnn.message_passing.nnconv mlreco.models.layers.gnn.message_passing.nnconv_elu mlreco.models.layers.gnn.message_passing.nnconv_old mlreco.models.layers.gnn.normalizations mlreco.models.singlep mlreco.models.spice mlreco.models.uresnet mlreco.models.uresnet_ppn_chain mlreco.post_processing mlreco.post_processing.analysis mlreco.post_processing.analysis.instance_clustering mlreco.post_processing.analysis.michel_reconstruction mlreco.post_processing.analysis.michel_reconstruction_2d mlreco.post_processing.analysis.michel_reconstruction_noghost mlreco.post_processing.analysis.muon_residual_range mlreco.post_processing.analysis.nue_selection mlreco.post_processing.analysis.stopping_muons mlreco.post_processing.analysis.through_muons mlreco.post_processing.analysis.track_clustering mlreco.post_processing.common mlreco.post_processing.decorator mlreco.post_processing.metrics mlreco.post_processing.metrics.bayes_segnet_mcdropout mlreco.post_processing.metrics.cluster_cnn_metrics mlreco.post_processing.metrics.cluster_gnn_metrics mlreco.post_processing.metrics.cosmic_discriminator_metrics mlreco.post_processing.metrics.deghosting_metrics mlreco.post_processing.metrics.duq_metrics mlreco.post_processing.metrics.evidential_gnn mlreco.post_processing.metrics.evidential_metrics mlreco.post_processing.metrics.evidential_segnet mlreco.post_processing.metrics.graph_spice_metrics mlreco.post_processing.metrics.kinematics_metrics mlreco.post_processing.metrics.pid_metrics mlreco.post_processing.metrics.ppn_metrics mlreco.post_processing.metrics.ppn_simple mlreco.post_processing.metrics.single_particle mlreco.post_processing.metrics.singlep_mcdropout mlreco.post_processing.metrics.uresnet_metrics mlreco.post_processing.metrics.vertex_metrics mlreco.post_processing.store mlreco.post_processing.store.store_input mlreco.post_processing.store.store_output mlreco.post_processing.store.store_uresnet mlreco.post_processing.store.store_uresnet_ppn mlreco.trainval mlreco.utils mlreco.utils.adabound mlreco.utils.data_parallel mlreco.utils.dbscan mlreco.utils.deghosting mlreco.utils.groups mlreco.utils.metrics mlreco.utils.numba mlreco.utils.ppn mlreco.utils.track_clustering mlreco.utils.unwrap mlreco.utils.utils mlreco.utils.vertex mlreco.visualization mlreco.visualization.cubes mlreco.visualization.gnn mlreco.visualization.plotly_layouts mlreco.visualization.points mlreco.visualization.voxels MODULE (mlreco.models.layers.common.nonlinearities.MinkowskiELU attribute) (mlreco.models.layers.common.nonlinearities.MinkowskiLeakyReLU attribute) MODULES (mlreco.models.bayes_uresnet.BayesianUResNet attribute) (mlreco.models.bayes_uresnet.DUQUResNet attribute) (mlreco.models.full_chain.FullChain attribute) (mlreco.models.graph_spice.MinkGraphSPICE attribute) (mlreco.models.grappa.GNN attribute) (mlreco.models.layers.cluster_cnn.graph_spice_embedder.GraphSPICEEmbedder attribute) (mlreco.models.layers.common.gnn_full_chain.FullChainGNN attribute) (mlreco.models.singlep.BayesianParticleClassifier attribute) (mlreco.models.singlep.DUQParticleClassifier attribute) (mlreco.models.singlep.EvidentialParticleClassifier attribute) (mlreco.models.singlep.ParticleImageClassifier attribute) (mlreco.models.spice.MinkSPICE attribute) (mlreco.models.uresnet.UResNet_Chain attribute) (mlreco.models.uresnet_ppn_chain.UResNetPPN attribute) momentum (mlreco.models.layers.gnn.normalizations.BatchNorm attribute) (mlreco.models.layers.gnn.normalizations.InstanceNorm attribute) MomentumNet (class in mlreco.models.layers.common.momentum) moving_average() (in module mlreco.utils.utils) MultiLabelCrossEntropy (class in mlreco.models.singlep) multivariate_kernel() (in module mlreco.models.layers.cluster_cnn.losses.misc) MultiVariateLovasz (class in mlreco.models.layers.cluster_cnn.losses.spatial_embeddings) muon_residual_range() (in module mlreco.post_processing.analysis.muon_residual_range) must_invert() (in module analysis.algorithms.selections.through_going_muons) mvgauss() (in module mlreco.models.layers.gnn.cluster)

N

network_schematic() (in module mlreco.visualization.gnn) network_topology() (in module mlreco.visualization.gnn) nms_numpy() (in module mlreco.post_processing.store.store_uresnet_ppn) (in module mlreco.utils.ppn) NNConvModel (class in mlreco.models.layers.gnn.message_passing.nnconv) (class in mlreco.models.layers.gnn.message_passing.nnconv_elu) (class in mlreco.models.layers.gnn.message_passing.nnconv_old) node_encoder_construct() (in module mlreco.models.layers.gnn.factories) node_encoder_dict() (in module mlreco.models.layers.gnn.factories) node_loss_construct() (in module mlreco.models.layers.gnn.factories) node_loss_dict() (in module mlreco.models.layers.gnn.factories) NodeEdgeHybridLoss (class in mlreco.models.layers.cluster_cnn.losses.gs_embeddings) NodeEvidentialKinematicsLoss (class in mlreco.models.layers.gnn.losses.node_kinematics) NodeKinematicsLoss (class in mlreco.models.layers.gnn.losses.node_kinematics) NodeLayer (class in mlreco.models.layers.gnn.message_passing.agnnconv) (class in mlreco.models.layers.gnn.message_passing.econv) (class in mlreco.models.layers.gnn.message_passing.gatconv) (class in mlreco.models.layers.gnn.message_passing.meta) (class in mlreco.models.layers.gnn.message_passing.nnconv) (class in mlreco.models.layers.gnn.message_passing.nnconv_elu) (class in mlreco.models.layers.gnn.message_passing.nnconv_old)

NodePrimaryLoss (class in mlreco.models.layers.gnn.losses.node_primary) NodeTransformerLoss (class in mlreco.models.layers.gnn.losses.node_kinematics) NodeTypeLoss (class in mlreco.models.layers.gnn.losses.node_type) normalizations_construct() (in module mlreco.models.layers.common.activation_normalization_factories) normalizations_dict() (in module mlreco.models.layers.common.activation_normalization_factories) normalize_coords() (in module mlreco.models.layers.common.blocks) nue_selection() (in module mlreco.post_processing.analysis.nue_selection) num_features (mlreco.models.layers.gnn.normalizations.BatchNorm attribute) (mlreco.models.layers.gnn.normalizations.InstanceNorm attribute) num_pred_clusters() (in module mlreco.post_processing.metrics.graph_spice_metrics) num_small_clusters() (in module mlreco.post_processing.metrics.graph_spice_metrics) num_true_clusters() (in module mlreco.post_processing.metrics.graph_spice_metrics) num_volumes() (mlreco.iotools.collates.VolumeBoundaries method) numba_wrapper() (in module mlreco.utils.numba)

O

occupancy_loss() (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.GraphSPICEEmbeddingLoss method)

P

parse_cluster2d() (in module mlreco.iotools.parsers.cluster) parse_cluster3d() (in module mlreco.iotools.parsers.cluster) parse_cluster3d_charge_rescaled() (in module mlreco.iotools.parsers.cluster) parse_crthits() (in module mlreco.iotools.parsers.misc) parse_meta2d() (in module mlreco.iotools.parsers.misc) parse_meta3d() (in module mlreco.iotools.parsers.misc) parse_neutrinos() (in module mlreco.iotools.parsers.particles) parse_opflash() (in module mlreco.iotools.parsers.misc) parse_particle_coords() (in module mlreco.iotools.parsers.particles) parse_particle_graph() (in module mlreco.iotools.parsers.particles) parse_particle_points() (in module mlreco.iotools.parsers.particles) parse_particle_singlep_einit() (in module mlreco.iotools.parsers.particles) parse_particle_singlep_pdg() (in module mlreco.iotools.parsers.particles) parse_particles() (in module mlreco.iotools.parsers.particles) parse_run_info() (in module mlreco.iotools.parsers.misc) parse_sparse2d() (in module mlreco.iotools.parsers.sparse) parse_sparse3d() (in module mlreco.iotools.parsers.sparse) parse_sparse3d_charge_rescaled() (in module mlreco.iotools.parsers.sparse)

parse_sparse3d_ghost() (in module mlreco.iotools.parsers.sparse) ParticleImageClassifier (class in mlreco.models.singlep) ParticleTypeLoss (class in mlreco.models.singlep) pass_particle() (in module mlreco.utils.ppn) pid_metrics() (in module mlreco.post_processing.metrics.pid_metrics) plotly_layout3d() (in module mlreco.visualization.plotly_layouts) post_processing() (in module mlreco.post_processing.decorator) PPN (class in mlreco.models.layers.common.ppnplus) ppn_metrics() (in module mlreco.post_processing.metrics.ppn_metrics) ppn_simple() (in module mlreco.post_processing.metrics.ppn_simple) PPNLonelyLoss (class in mlreco.models.layers.common.ppnplus) predict_vertex() (in module mlreco.utils.vertex) prepare() (in module mlreco.main_funcs) print_memory() (in module mlreco.utils.utils) process_config() (in module mlreco.main_funcs) process_group_data() (in module mlreco.utils.groups) progress_bar() (in module mlreco.utils.utils) purity() (in module mlreco.utils.metrics) purity_efficiency() (in module mlreco.utils.metrics)

R

radius_neighbor_loss() (mlreco.models.layers.cluster_cnn.losses.radius_nnloss.DensityBasedNNLoss method) randomize_labels() (analysis.classes.ui.FullChainPredictor static method) RandomSequenceSampler (class in mlreco.iotools.samplers) record() (mlreco.utils.utils.ChunkCSVData method) (mlreco.utils.utils.CSVData method) recursive_merge() (in module analysis.decorator) reduction (mlreco.models.full_chain.FullChainLoss attribute) (mlreco.models.grappa.GNNLoss attribute) (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.NodeEdgeHybridLoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.LovaszHingeLoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.LovaszSoftmaxWithLogitsLoss attribute) (mlreco.models.layers.cluster_cnn.losses.radius_nnloss.DensityBasedNNLoss attribute) (mlreco.models.layers.common.gnn_full_chain.FullChainLoss attribute) (mlreco.models.layers.common.ppnplus.PPNLonelyLoss attribute) (mlreco.models.layers.gnn.losses.node_kinematics.BerHuLoss attribute) (mlreco.models.layers.gnn.losses.node_kinematics.LogRMSE attribute) (mlreco.models.uresnet.SegmentationLoss attribute)

regularization() (mlreco.models.layers.cluster_cnn.losses.single_layers.DiscriminativeLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterLoss method) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast.SPICEInterLoss method) regularization_loss() (in module mlreco.models.layers.cluster_cnn.losses.misc) ResNetBlock (class in mlreco.models.layers.common.blocks) ResNeXtBlock (class in mlreco.models.layers.common.blocks) round_decimals() (in module mlreco.utils.utils) run_fragment_gnns() (mlreco.models.layers.common.gnn_full_chain.FullChainGNN method) run_gnn() (mlreco.models.layers.common.gnn_full_chain.FullChainGNN method) run_particle_gnns() (mlreco.models.layers.common.gnn_full_chain.FullChainGNN method)

S

save_state() (mlreco.trainval.trainval method) SBD() (in module mlreco.utils.metrics) scatter() (mlreco.utils.data_parallel.DataParallel method) scatter_clusters() (in module mlreco.visualization.gnn) scatter_cubes() (in module mlreco.visualization.cubes) scatter_label() (in module mlreco.visualization.voxels) scatter_points() (in module mlreco.visualization.points) scatter_voxels() (in module mlreco.visualization.voxels) SEBlock (class in mlreco.models.layers.common.blocks) SegmentationLoss (class in mlreco.models.bayes_uresnet) (class in mlreco.models.uresnet) select_particle_in_group() (mlreco.models.layers.common.gnn_full_chain.FullChainGNN method) SENet (class in mlreco.models.layers.common.senet) SequentialBatchSampler (class in mlreco.iotools.samplers) SEResNetBlock (class in mlreco.models.layers.common.blocks) setup_chain_cfg() (in module mlreco.models.layers.common.gnn_full_chain) setup_cnn_configuration() (in module mlreco.models.layers.common.configuration) single_particle() (in module mlreco.post_processing.metrics.single_particle) singlep_mcdropout() (in module mlreco.post_processing.metrics.singlep_mcdropout) softmax_nb() (in module mlreco.utils.numba) SparseGenerator (class in mlreco.models.layers.common.sparse_generator) SparseGeneratorSimple (class in mlreco.models.layers.common.sparse_generator) SparseResidualEncoder (class in mlreco.models.layers.common.cnn_encoder) SparseToDense (class in mlreco.models.layers.common.blocks) spatial_embedding_loss() (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.GraphSPICEEmbeddingLoss method) SPICE (class in mlreco.models.layers.cluster_cnn.embeddings)

spice_loss_construct() (in module mlreco.models.layers.cluster_cnn.factories) spice_loss_dict() (in module mlreco.models.layers.cluster_cnn.factories) SPICEInterLoss (class in mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast) SPICELoss (class in mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast) (class in mlreco.models.spice) split() (mlreco.iotools.collates.VolumeBoundaries method) SPP (class in mlreco.models.layers.common.blocks) squared_distances() (in module mlreco.models.layers.cluster_cnn.losses.misc) StableBCELoss (class in mlreco.models.layers.cluster_cnn.losses.lovasz) standard_forward() (mlreco.models.bayes_uresnet.BayesianUResNet method) (mlreco.models.singlep.BayesianParticleClassifier method) start() (mlreco.utils.utils.stopwatch method) start_cputime() (mlreco.utils.utils.stopwatch method) step() (mlreco.utils.adabound.AdaBound method) (mlreco.utils.adabound.AdaBoundW method) stop() (mlreco.utils.utils.stopwatch method) stop_cputime() (mlreco.utils.utils.stopwatch method) stopping_muons() (in module analysis.algorithms.selections.stopping_muons) (in module mlreco.post_processing.analysis.stopping_muons) stopwatch (class in mlreco.utils.utils) store_input() (in module mlreco.post_processing.store.store_input) store_output() (in module mlreco.post_processing.store.store_output) store_uresnet() (in module mlreco.post_processing.store.store_uresnet) store_uresnet_ppn() (in module mlreco.post_processing.store.store_uresnet_ppn) student_t() (in module mlreco.models.layers.gnn.cluster) submatrix_nb() (in module mlreco.utils.numba)

T

through_going_muons() (in module analysis.algorithms.selections.through_going_muons) through_muons() (in module mlreco.post_processing.analysis.through_muons) time() (mlreco.utils.utils.stopwatch method) time_cputime() (mlreco.utils.utils.stopwatch method) to_numpy() (in module mlreco.utils.utils) to_sparse() (in module mlreco.models.layers.common.blocks) trace_interactions() (in module mlreco.visualization.plotly_layouts) trace_particles() (in module mlreco.visualization.plotly_layouts) track_clustering() (in module mlreco.post_processing.analysis.track_clustering) (in module mlreco.utils.track_clustering) track_running_stats (mlreco.models.layers.gnn.normalizations.BatchNorm attribute) (mlreco.models.layers.gnn.normalizations.InstanceNorm attribute) train() (in module mlreco.main_funcs) train_logger (mlreco.main_funcs.Handlers attribute) train_loop() (in module mlreco.main_funcs) train_step() (mlreco.trainval.trainval method) training (mlreco.models.bayes_uresnet.BayesianUResNet attribute) (mlreco.models.bayes_uresnet.DUQSegmentationLoss attribute) (mlreco.models.bayes_uresnet.DUQUResNet attribute) (mlreco.models.bayes_uresnet.SegmentationLoss attribute) (mlreco.models.full_chain.FullChain attribute) (mlreco.models.graph_spice.GraphSPICELoss attribute) (mlreco.models.graph_spice.MinkGraphSPICE attribute) (mlreco.models.grappa.GNN attribute) (mlreco.models.layers.cluster_cnn.embeddings.Attention attribute) (mlreco.models.layers.cluster_cnn.embeddings.ExpandAs attribute) (mlreco.models.layers.cluster_cnn.embeddings.SPICE attribute) (mlreco.models.layers.cluster_cnn.graph_spice_embedder.GraphSPICEEmbedder attribute) (mlreco.models.layers.cluster_cnn.gs_kernels.BilinearKernel attribute) (mlreco.models.layers.cluster_cnn.gs_kernels.BilinearNNKernel attribute) (mlreco.models.layers.cluster_cnn.gs_kernels.DefaultKernel attribute) (mlreco.models.layers.cluster_cnn.gs_kernels.MixedKernel attribute) (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.GraphSPICEEmbeddingLoss attribute) (mlreco.models.layers.cluster_cnn.losses.gs_embeddings.WeightedEdgeLoss attribute) (mlreco.models.layers.cluster_cnn.losses.lovasz.StableBCELoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.BinaryCELogDiceLoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.BinaryLogDiceLoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.FocalLoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.IoUScore attribute) (mlreco.models.layers.cluster_cnn.losses.misc.MincutLoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.VectorEstimationLoss attribute) (mlreco.models.layers.cluster_cnn.losses.misc.WeightedFocalLoss attribute) (mlreco.models.layers.cluster_cnn.losses.single_layers.DiscriminativeLoss attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.CELovaszLoss attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELoss attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELoss2 attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskBCELossBivariate attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskFocalLoss attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszHingeLoss attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterBC attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterLoss attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MaskLovaszInterLoss2 attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings.MultiVariateLovasz attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast.SPICEInterLoss attribute) (mlreco.models.layers.cluster_cnn.losses.spatial_embeddings_fast.SPICELoss attribute) (mlreco.models.layers.common.blocks.DenseResBlock attribute) (mlreco.models.layers.common.blocks.Identity attribute) (mlreco.models.layers.common.blocks.SparseToDense attribute) (mlreco.models.layers.common.cluster_pool.ClusterPool attribute) (mlreco.models.layers.common.cnn_encoder.SparseResidualEncoder attribute) (mlreco.models.layers.common.dbscan.DBSCANFragmenter attribute) (mlreco.models.layers.common.extract_feature_map.MinkGhostMask attribute) (mlreco.models.layers.common.fpn.FPN attribute) (mlreco.models.layers.common.gnn_full_chain.FullChainGNN attribute) (mlreco.models.layers.common.mobilenet.MB3Encoder attribute) (mlreco.models.layers.common.mobilenet.MobileNetV3 attribute) (mlreco.models.layers.common.momentum.DeepVertexNet attribute) (mlreco.models.layers.common.momentum.EvidentialMomentumNet attribute) (mlreco.models.layers.common.momentum.MomentumNet attribute) (mlreco.models.layers.common.momentum.VertexNet attribute) (mlreco.models.layers.common.nonlinearities.MinkowskiMish attribute) (mlreco.models.layers.common.normalizations.MinkowskiAdaIN attribute) (mlreco.models.layers.common.normalizations.MinkowskiGroupNorm attribute) (mlreco.models.layers.common.normalizations.MinkowskiPixelNorm attribute) (mlreco.models.layers.common.ppnplus.AttentionMask attribute) (mlreco.models.layers.common.ppnplus.ExpandAs attribute) (mlreco.models.layers.common.ppnplus.MergeConcat attribute) (mlreco.models.layers.common.ppnplus.PPN attribute) (mlreco.models.layers.common.senet.SENet attribute) (mlreco.models.layers.common.sparse_generator.SparseGenerator attribute) (mlreco.models.layers.common.sparse_generator.SparseGeneratorSimple attribute) (mlreco.models.layers.common.uresnet_layers.UResNet attribute) (mlreco.models.layers.common.uresnet_layers.UResNetDecoder attribute) (mlreco.models.layers.common.uresnet_layers.UResNetEncoder attribute) (mlreco.models.layers.common.uresnext.UResNeXt attribute) (mlreco.models.layers.gnn.encoders.cnn.ClustCNNMinkEdgeEncoder attribute) (mlreco.models.layers.gnn.encoders.cnn.ClustCNNMinkNodeEncoder attribute) (mlreco.models.layers.gnn.encoders.geometric.ClustGeoEdgeEncoder attribute) (mlreco.models.layers.gnn.encoders.geometric.ClustGeoNodeEncoder attribute) (mlreco.models.layers.gnn.encoders.mixed.ClustMixEdgeEncoder attribute) (mlreco.models.layers.gnn.encoders.mixed.ClustMixNodeEncoder attribute) (mlreco.models.layers.gnn.losses.edge_channel.EdgeChannelLoss attribute) (mlreco.models.layers.gnn.losses.node_grouping.GNNGroupingLoss attribute) (mlreco.models.layers.gnn.losses.node_kinematics.NodeEvidentialKinematicsLoss attribute) (mlreco.models.layers.gnn.losses.node_kinematics.NodeKinematicsLoss attribute) (mlreco.models.layers.gnn.losses.node_kinematics.NodeTransformerLoss attribute) (mlreco.models.layers.gnn.losses.node_primary.NodePrimaryLoss attribute) (mlreco.models.layers.gnn.losses.node_type.NodeTypeLoss attribute) (mlreco.models.layers.gnn.message_passing.agnnconv.AGNNConvModel attribute) (mlreco.models.layers.gnn.message_passing.agnnconv.EdgeLayer attribute) (mlreco.models.layers.gnn.message_passing.agnnconv.GlobalModel attribute) (mlreco.models.layers.gnn.message_passing.agnnconv.NodeLayer attribute) (mlreco.models.layers.gnn.message_passing.dgtransformer.DeepGraphTransformer attribute) (mlreco.models.layers.gnn.message_passing.econv.EConvModel attribute) (mlreco.models.layers.gnn.message_passing.econv.EdgeLayer attribute) (mlreco.models.layers.gnn.message_passing.econv.GlobalModel attribute) (mlreco.models.layers.gnn.message_passing.econv.NodeLayer attribute) (mlreco.models.layers.gnn.message_passing.gatconv.EdgeLayer attribute) (mlreco.models.layers.gnn.message_passing.gatconv.GATConvModel attribute) (mlreco.models.layers.gnn.message_passing.gatconv.GlobalModel attribute) (mlreco.models.layers.gnn.message_passing.gatconv.NodeLayer attribute) (mlreco.models.layers.gnn.message_passing.meta.EdgeLayer attribute) (mlreco.models.layers.gnn.message_passing.meta.GlobalModel attribute) (mlreco.models.layers.gnn.message_passing.meta.MetaLayerModel attribute) (mlreco.models.layers.gnn.message_passing.meta.NodeLayer attribute) (mlreco.models.layers.gnn.message_passing.nnconv.EdgeLayer attribute) (mlreco.models.layers.gnn.message_passing.nnconv.GlobalModel attribute) (mlreco.models.layers.gnn.message_passing.nnconv.NNConvModel attribute) (mlreco.models.layers.gnn.message_passing.nnconv.NodeLayer attribute) (mlreco.models.layers.gnn.message_passing.nnconv_elu.EdgeLayer attribute) (mlreco.models.layers.gnn.message_passing.nnconv_elu.GlobalModel attribute) (mlreco.models.layers.gnn.message_passing.nnconv_elu.NNConvModel attribute) (mlreco.models.layers.gnn.message_passing.nnconv_elu.NodeLayer attribute) (mlreco.models.layers.gnn.message_passing.nnconv_old.EdgeLayer attribute) (mlreco.models.layers.gnn.message_passing.nnconv_old.GlobalModel attribute) (mlreco.models.layers.gnn.message_passing.nnconv_old.NNConvModel attribute) (mlreco.models.layers.gnn.message_passing.nnconv_old.NodeLayer attribute) (mlreco.models.singlep.BayesianParticleClassifier attribute) (mlreco.models.singlep.DUQParticleClassifier attribute) (mlreco.models.singlep.EvidentialLearningLoss attribute) (mlreco.models.singlep.EvidentialParticleClassifier attribute) (mlreco.models.singlep.MultiLabelCrossEntropy attribute) (mlreco.models.singlep.ParticleImageClassifier attribute) (mlreco.models.singlep.ParticleTypeLoss attribute) (mlreco.models.spice.MinkSPICE attribute) (mlreco.models.spice.SPICELoss attribute) (mlreco.models.uresnet.UResNet_Chain attribute) (mlreco.models.uresnet_ppn_chain.UResNetPPN attribute) (mlreco.models.uresnet_ppn_chain.UResNetPPNLoss attribute) (mlreco.utils.data_parallel.DataParallel attribute)

trainval (class in mlreco.trainval) translate() (mlreco.iotools.collates.VolumeBoundaries method)

U

unique_label() (in module mlreco.utils.metrics) unique_label_torch() (in module mlreco.models.layers.cluster_cnn.losses.misc) unique_nb() (in module mlreco.utils.numba) unique_with_batch() (in module mlreco.utils.metrics) untranslate() (mlreco.iotools.collates.VolumeBoundaries method) unwrap() (in module mlreco.utils.unwrap) unwrap_2d_scn() (in module mlreco.utils.unwrap) unwrap_3d_mink() (in module mlreco.utils.unwrap) unwrap_3d_scn() (in module mlreco.utils.unwrap) unwrap_scn() (in module mlreco.utils.unwrap) update_buffers() (mlreco.models.bayes_uresnet.DUQUResNet method) (mlreco.models.singlep.DUQParticleClassifier method)

update_dict() (mlreco.utils.utils.ForwardData method) update_mean() (mlreco.utils.utils.ForwardData method) UResNet (class in mlreco.models.layers.common.uresnet_layers) UResNet_Chain (class in mlreco.models.uresnet) uresnet_metrics() (in module mlreco.post_processing.metrics.uresnet_metrics) uresnet_ppn_point_selector() (in module mlreco.utils.ppn) uresnet_ppn_type_point_selector() (in module mlreco.utils.ppn) UResNetDecoder (class in mlreco.models.layers.common.uresnet_layers) UResNetEncoder (class in mlreco.models.layers.common.uresnet_layers) UResNetPPN (class in mlreco.models.uresnet_ppn_chain) UResNetPPNLoss (class in mlreco.models.uresnet_ppn_chain) UResNeXt (class in mlreco.models.layers.common.uresnext)

V

VectorEstimationLoss (class in mlreco.models.layers.cluster_cnn.losses.misc) vertex_metrics() (in module mlreco.post_processing.metrics.vertex_metrics)

VertexNet (class in mlreco.models.layers.common.momentum) virtual_batch_ids() (mlreco.iotools.collates.VolumeBoundaries method) VolumeBoundaries (class in mlreco.iotools.collates)

W

watch (mlreco.main_funcs.Handlers attribute) weight() (mlreco.models.layers.common.normalizations.MinkowskiAdaIN property) weight_initialization() (mlreco.models.graph_spice.MinkGraphSPICE method) weight_io (mlreco.main_funcs.Handlers attribute)

WeightedEdgeLoss (class in mlreco.models.layers.cluster_cnn.losses.gs_embeddings) WeightedFocalLoss (class in mlreco.models.layers.cluster_cnn.losses.misc) white_layout() (in module mlreco.visualization.plotly_layouts) write() (mlreco.utils.utils.CSVData method)

X

xloss() (in module mlreco.models.layers.cluster_cnn.losses.lovasz)

Y

yz_calibrations() (in module analysis.algorithms.selections.michel_electrons)