Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-02-06 03:32:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixing bug in open source implementation of OGB-LSC/PCQ.

Browse Source 
PiperOrigin-RevId: 483319710
This commit is contained in:
Alvaro Sanchez-Gonzalez
2022-10-24 09:29:40 +00:00
committed by Diego de las Casas
parent a748a7c817
commit a318c69018
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
ogb_lsc/pcq/dataset_utils.py
View File

@@ -289,6 +289,7 @@ def _load_smiles(
elif split == "train":
indices = datasets.load_all_except_kth_fold_indices(
data_root, k_fold_split_id, num_k_fold_splits)
indices += datasets.load_splits()["train"]
else:
assert split == "valid"
indices = datasets.load_kth_fold_indices(data_root, k_fold_split_id)