InferenceConfig Namespace Reference

Functions
	GraphInferenceAlgCfg (flags, name="GraphInferenceAlg", **kwargs)
	GraphSPFilterToolCfg (flags, name="GraphSPFilterTool", **kwargs)
	GraphBucketFilterToolCfg (flags, name="GraphBucketFilterTool", **kwargs)

Function Documentation

GraphBucketFilterToolCfg()

InferenceConfig.GraphBucketFilterToolCfg	(		flags,
			name = "GraphBucketFilterTool",
		**	kwargs )

Definition at line 24 of file InferenceConfig.py.

def GraphBucketFilterToolCfg(flags, name ="GraphBucketFilterTool", **kwargs):
 
    from AthOnnxComps.OnnxRuntimeSessionConfig import OnnxRuntimeSessionToolCfg
 
    result = ComponentAccumulator()
    model_path = kwargs.pop("ModelPath", "edgecnn_multi_bucket_sparse_meta.onnx")
    
    if not model_path.startswith('/'):
        pass
    else:
        pass
    
    kwargs.setdefault("ModelSession", result.popToolsAndMerge(OnnxRuntimeSessionToolCfg(flags, model_fname=model_path)))
    # BiasClass0: Working point selection bias for multi-class comparison
    # Higher values make class 0 (reject) less likely, accepting more buckets
    # Typical values: 0.5-2.0. Default: 1.0 (no bias)
    kwargs.setdefault("BiasClass0", 1.0)
    kwargs.setdefault("OutputLevel", 3)  # INFO level (1=VERBOSE, 2=DEBUG, 3=INFO, 4=WARNING, 5=ERROR, 6=FATAL)
 
    the_tool = CompFactory.MuonML.GraphBucketFilterTool(name, **kwargs)
    result.setPrivateTools(the_tool)
    return result
 

GraphInferenceAlgCfg()

InferenceConfig.GraphInferenceAlgCfg	(		flags,
			name = "GraphInferenceAlg",
		**	kwargs )

Definition at line 6 of file InferenceConfig.py.

def GraphInferenceAlgCfg(flags, name = "GraphInferenceAlg", **kwargs):
    result  = ComponentAccumulator()
    the_alg = CompFactory.MuonML.InferenceAlg(name, **kwargs)
    result.addEventAlgo(the_alg, primary = True)
    return result
 

GraphSPFilterToolCfg()

InferenceConfig.GraphSPFilterToolCfg	(		flags,
			name = "GraphSPFilterTool",
		**	kwargs )

Definition at line 12 of file InferenceConfig.py.

def GraphSPFilterToolCfg(flags, name ="GraphSPFilterTool", **kwargs):
    
    from AthOnnxComps.OnnxRuntimeSessionConfig import OnnxRuntimeSessionToolCfg
 
    result = ComponentAccumulator()
    kwargs.setdefault("ModelSession", result.popToolsAndMerge(OnnxRuntimeSessionToolCfg(flags, model_fname="/eos/atlas/atlascerngroupdisk/data-art/grid-input/MuonRecRTT/TestModel.onnx")))
    kwargs.setdefault("MLFilterCut", -3.6) # Working point cut
    
    the_tool = CompFactory.MuonML.GraphSPFilterTool(name, **kwargs)
    result.setPrivateTools(the_tool)
    return result