nubench

IceCube-specific Detector class(es).

class graphnet.models.detector.nubench.NuBenchDetector(*args, **kwargs)

Bases: Detector

Generic Detector Class from the NuBench Paper (arXiv:2511.13111).

Construct Detector.

Parameters:

• replace_with_identity (Optional[List[str]], default: None) – A list of feature names from the

• identity (feature_map that should be replaced with the)

• function.

• args (Any)

• kwargs (Any)

Return type:

object

xyz = ['sensor_pos_x', 'sensor_pos_y', 'sensor_pos_z']

string_id_column = 'sensor_string_id'

sensor_id_column = 'sensor_id'

sensor_time_column = 't'

charge_column = 'charge'

feature_map()

Map standardization functions to each dimension.

Return type:

Dict[str, Callable]

class graphnet.models.detector.nubench.FlowerS(*args, **kwargs)

Bases: NuBenchDetector

Detector class for Flower S.

Construct Detector.

Parameters:

• replace_with_identity (Optional[List[str]], default: None) – A list of feature names from the

• identity (feature_map that should be replaced with the)

• function.

• args (Any)

• kwargs (Any)

Return type:

object

geometry_table_path = '/home/runner/work/graphnet/graphnet/data/geometry_tables/nubench/flower_s.parquet'

class graphnet.models.detector.nubench.FlowerL(*args, **kwargs)

Bases: NuBenchDetector

Detector class for Flower L.

Construct Detector.

Parameters:

• replace_with_identity (Optional[List[str]], default: None) – A list of feature names from the

• identity (feature_map that should be replaced with the)

• function.

• args (Any)

• kwargs (Any)

Return type:

object

geometry_table_path = '/home/runner/work/graphnet/graphnet/data/geometry_tables/nubench/flower_l.parquet'

class graphnet.models.detector.nubench.FlowerXL(*args, **kwargs)

Bases: NuBenchDetector

Detector class for Flower XL.

Construct Detector.

Parameters:

• replace_with_identity (Optional[List[str]], default: None) – A list of feature names from the

• identity (feature_map that should be replaced with the)

• function.

• args (Any)

• kwargs (Any)

Return type:

object

geometry_table_path = '/home/runner/work/graphnet/graphnet/data/geometry_tables/nubench/flower_xl.parquet'

class graphnet.models.detector.nubench.Triangle(*args, **kwargs)

Bases: NuBenchDetector

Detector class for Triangle.

Construct Detector.

Parameters:

• replace_with_identity (Optional[List[str]], default: None) – A list of feature names from the

• identity (feature_map that should be replaced with the)

• function.

• args (Any)

• kwargs (Any)

Return type:

object

geometry_table_path = '/home/runner/work/graphnet/graphnet/data/geometry_tables/nubench/triangle.parquet'

class graphnet.models.detector.nubench.Cluster(*args, **kwargs)

Bases: NuBenchDetector

Detector class for Cluster.

Construct Detector.

Parameters:

• replace_with_identity (Optional[List[str]], default: None) – A list of feature names from the

• identity (feature_map that should be replaced with the)

• function.

• args (Any)

• kwargs (Any)

Return type:

object

geometry_table_path = '/home/runner/work/graphnet/graphnet/data/geometry_tables/nubench/cluster.parquet'

class graphnet.models.detector.nubench.Hexagon(*args, **kwargs)

Bases: NuBenchDetector

Detector class for Hexagon.

Construct Detector.

Parameters:

• replace_with_identity (Optional[List[str]], default: None) – A list of feature names from the

• identity (feature_map that should be replaced with the)

• function.

• args (Any)

• kwargs (Any)

Return type:

object

geometry_table_path = '/home/runner/work/graphnet/graphnet/data/geometry_tables/nubench/hexagon.parquet'