dataset_name example for csce ORNL#319

kshitij-v-mehta · Jan 27, 2025 · 8fca19f · 8fca19f
1 parent edafeb5
commit 8fca19f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/csce/train_gap.py b/examples/csce/train_gap.py
@@ -288,6 +288,7 @@ def __getitem__(self, idx):
                 data = generate_graphdata_from_smilestr(
                     smilestr, ytarget, csce_node_types, var_config
                 )
+                data.dataset_name = 'csce'
                 dataset_lists[idataset].append(data)
 
         trainset = dataset_lists[0]