Update sampl_graph_conv.py

"""

Script that trains graph-conv models on Tox21 dataset.

Script that trains graph-conv models on SAMPL(FreeSolv) dataset.

"""

from __future__ import print_function

from __future__ import division

import deepchem as dc

from deepchem.molnet import load_sampl

# Load Tox21 dataset

# Load SAMPL(FreeSolv) dataset

SAMPL_tasks, SAMPL_datasets, transformers = load_sampl(featurizer='GraphConv')

train_dataset, valid_dataset, test_dataset = SAMPL_datasets