Commit 1d002eef authored by miaecle's avatar miaecle
Browse files

small fix

parent ff608daf
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -13,7 +13,7 @@ import deepchem as dc

# Load tox21 dataset
tox21_tasks, tox21_datasets, transformers = dc.molnet.load_tox21(
    featurizer='Weave', split='index')
    featurizer='GraphConv', split='index')
train_dataset, valid_dataset, test_dataset = tox21_datasets

# Fit models