VoronoiGNNDataset Class Reference

Dataset for non-hybrid GNN training. More...

Inheritance diagram for VoronoiGNNDataset:

Public Member Functions
	__init__ (self, data_dir, stage, output_dim)
	len (self)
	get (self, idx)

Public Attributes
	stage = stage
	output_dim = output_dim
	features = DataLoaderUtils.load_features(f"{data_dir}/{stage}", output_dim)
	dataset_size = self.features[0].shape[0]
	targets
	targets_mean
tuple	targets_std
	edge_weights = DataLoaderUtils.load_edge_weights(f"{data_dir}/{stage}")

Detailed Description

Dataset for non-hybrid GNN training.

Definition at line 148 of file loaders.py.

Constructor & Destructor Documentation

__init__()

__init__	(		self,
			data_dir,
			stage,
			output_dim )

Definition at line 150 of file loaders.py.

Member Function Documentation

get()

get	(		self,
			idx )

Definition at line 166 of file loaders.py.

len()

len

(

self

)

Definition at line 163 of file loaders.py.

Member Data Documentation

dataset_size

dataset_size = self.features[0].shape[0]

Definition at line 157 of file loaders.py.

edge_weights

edge_weights = DataLoaderUtils.load_edge_weights(f"{data_dir}/{stage}")

Definition at line 161 of file loaders.py.

features

features = DataLoaderUtils.load_features(f"{data_dir}/{stage}", output_dim)

Definition at line 156 of file loaders.py.

output_dim

output_dim = output_dim

Definition at line 154 of file loaders.py.

stage

stage = stage

Definition at line 153 of file loaders.py.

targets

targets

Definition at line 158 of file loaders.py.

targets_mean

targets_mean

Definition at line 158 of file loaders.py.

targets_std

tuple targets_std

Initial value:

=  (
                DataLoaderUtils.load_actions(f"{data_dir}/{stage}")
                )

Definition at line 158 of file loaders.py.