Skip to content

Commit 3984b9a

Browse files
committed
update dataset load output and add zinc250k
1 parent 2eadef7 commit 3984b9a

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -96,6 +96,7 @@ molecular_data = load_qm9(local_dir='torchmol_data', target_cols=target_cols)
9696
smiles_list, property_np_array = molecular_data.data, molecular_data.target
9797

9898
# the target could be None if loading an unlabeled dataset
99+
from torch_molecule.datasets import load_zinc250k
99100
molecular_data = load_zinc250k(local_dir='torchmol_data', target_cols=target_cols)
100101
smiles_list = molecular_data.data
101102
assert molecular_data.target is None

0 commit comments

Comments
 (0)