graphorge.gnn_base_model.data.graph_dataset.get_dataset_sample_files_from_dir

get_dataset_sample_files_from_dir(dataset_directory, sample_file_basename)[source]

Get Graph Neural Network graph data set samples files from directory.

Parameters:
  • dataset_directory (str) – Directory where the Graph Neural Network graph data set is stored (all data set samples files).

  • sample_file_basename (str) – Basename of Graph Neural Network graph data set sample file. The basename is appended with sample index.

Returns:

dataset_sample_files – Graph Neural Network graph data set samples files paths. Each sample file contains a torch_geometric.data.Data object describing a homogeneous graph.

Return type:

list[str]