graphorge.gnn_base_model.data.graph_dataset.write_graph_dataset_summary_file

write_graph_dataset_summary_file(dataset_directory, n_sample, total_time_sec, avg_time_sample, node_features=None, edge_features=None, global_features=None, node_targets=None, edge_targets=None, global_targets=None, filename='summary')[source]

Write summary data file for GNN-based data set generation.

Parameters:

  • dataset_directory (str) – Directory where the Graph Neural Network graph data set is stored (all data set samples files). All existent files are overridden when saving sample data files.

  • n_sample (int) – Data set size (number of samples).

  • total_time_sec (int) – Total generation time in seconds.

  • avg_time_sample (float) – Average generation time per sample.

  • node_features (tuple[str], default=None) – Graph Neural Network graph data set nodes features.

  • edge_features (tuple[str], default=None) – Graph Neural Network graph data set edges features.

  • global_features (tuple[str], default=None) – Graph Neural Network graph data set global features.

  • node_targets (tuple[str], default=None) – Graph Neural Network graph data set nodes targets.

  • edge_targets (tuple[str], default=None) – Graph Neural Network graph data set edges targets.

  • global_targets (tuple[str], default=None) – Graph Neural Network graph data set global targets.

  • filename (str, default='summary') – Summary file name.