Skip to content

Commit

Permalink
Blank change
Browse files Browse the repository at this point in the history
  • Loading branch information
Gaurav S Deshmukh committed Sep 21, 2023
1 parent e98d998 commit ffac10b
Showing 1 changed file with 11 additions and 11 deletions.
22 changes: 11 additions & 11 deletions src/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -141,15 +141,15 @@ def get(self, i):

# Create dataset
dataset = AtomsDataset(data_root_path, prop_csv_path)
dataset.process_data(z_cutoffs=[13., 20.],
node_features=[
["atomic_number", "dband_center"],
["atomic_number", "reactivity"],
["atomic_number", "reactivity"],
],
edge_features=[
["bulk_bond_distance"],
["surface_bond_distance"],
["adsorbate_bond_distance"],
])
# dataset.process_data(z_cutoffs=[13., 20.],
# node_features=[
# ["atomic_number", "dband_center"],
# ["atomic_number", "reactivity"],
# ["atomic_number", "reactivity"],
# ],
# edge_features=[
# ["bulk_bond_distance"],
# ["surface_bond_distance"],
# ["adsorbate_bond_distance"],
# ])
print(dataset[0][-2].x)

0 comments on commit ffac10b

Please sign in to comment.