CNNGNNDataset Class Reference

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

Inheritance diagram for CNNGNNDataset:

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

Public Attributes
	stage
	maps
	dataset_size
	targets
	targets_mean
	targets_std
	edge_weights
	robot_positions

Detailed Description

Dataset for hybrid CNN-GNN training.

Definition at line 170 of file loaders.py.

Constructor & Destructor Documentation

__init__()

__init__	(		self,
			data_dir,
			stage,
			use_comm_map,
			world_size )

Definition at line 172 of file loaders.py.

Member Function Documentation

get()

get	(		self,
			idx )

Definition at line 205 of file loaders.py.

len()

len

(

self

)

Definition at line 202 of file loaders.py.

Member Data Documentation

dataset_size

dataset_size

Definition at line 178 of file loaders.py.

edge_weights

edge_weights

Definition at line 183 of file loaders.py.

maps

maps

Definition at line 177 of file loaders.py.

robot_positions

robot_positions

Definition at line 185 of file loaders.py.

stage

stage

Definition at line 175 of file loaders.py.

targets

targets

Definition at line 180 of file loaders.py.

targets_mean

targets_mean

Definition at line 180 of file loaders.py.

targets_std

targets_std

Definition at line 180 of file loaders.py.