:py:mod:`n2j.trainval_data.graphs.base_graph` ============================================= .. py:module:: n2j.trainval_data.graphs.base_graph .. autoapi-nested-parse:: Generic catalog-agnostic module for input graph X Module Contents --------------- Classes ~~~~~~~ .. autoapisummary:: n2j.trainval_data.graphs.base_graph.Subgraph n2j.trainval_data.graphs.base_graph.BaseGraph .. py:class:: Subgraph(x=None, y_local=None, y=None, x_meta=None, y_class=None, edge_index=None) Bases: :py:obj:`torch_geometric.data.Data` Subgraph representing a single sightline .. py:class:: BaseGraph(root, raytracing_out_dir, aperture_size, n_data, debug=False, transform=None, pre_transform=None, pre_filter=None) Bases: :py:obj:`torch_geometric.data.Dataset` ABC for graphs created from photometric catalogs. Not to be used on its own. Child classes follow the naming convention, `Graph` .. py:method:: _get_sightlines() Load the precomputed sightlines containing the pointings and labels