n2j.trainval_data.graphs.base_graph#

Generic catalog-agnostic module for input graph X

Module Contents#

Classes#

Subgraph

Subgraph representing a single sightline

BaseGraph

ABC for graphs created from photometric catalogs. Not to be used on its
class n2j.trainval_data.graphs.base_graph.Subgraph(x=None, y_local=None, y=None, x_meta=None, y_class=None, edge_index=None)[source]#

Bases: torch_geometric.data.Data

Subgraph representing a single sightline

class n2j.trainval_data.graphs.base_graph.BaseGraph(root, raytracing_out_dir, aperture_size, n_data, debug=False, transform=None, pre_transform=None, pre_filter=None)[source]#

Bases: torch_geometric.data.Dataset

ABC for graphs created from photometric catalogs. Not to be used on its own. Child classes follow the naming convention, <name of catalog>Graph

_get_sightlines()[source]#

Load the precomputed sightlines containing the pointings and labels