Commit 1532918c authored by mufeili's avatar mufeili
Browse files

Update

parent 9d3fca5d
Loading
Loading
Loading
Loading
+10 −9
Original line number Diff line number Diff line
@@ -207,15 +207,16 @@ class GCNModel(TorchModel):
    Examples
    --------

    >>> import deepchem as dc
    >>> from deepchem.models import GCNModel
    >>> featurizer = dc.feat.MolGraphConvFeaturizer()
    >>> tasks, datasets, transformers = dc.molnet.load_tox21(
    ...     reload=False, featurizer=featurizer, transformers=[])
    >>> train, valid, test = datasets
    >>> model = dc.models.GCNModel(mode='classification', n_tasks=len(tasks),
    ...                            number_atom_features=30, batch_size=32, learning_rate=0.001)
    >>> model.fit(train, nb_epoch=50)
    >>>
    >> import deepchem as dc
    >> from deepchem.models import GCNModel
    >> featurizer = dc.feat.MolGraphConvFeaturizer()
    >> tasks, datasets, transformers = dc.molnet.load_tox21(
    ..     reload=False, featurizer=featurizer, transformers=[])
    >> train, valid, test = datasets
    >> model = dc.models.GCNModel(mode='classification', n_tasks=len(tasks),
    ..                            number_atom_features=30, batch_size=32, learning_rate=0.001)
    >> model.fit(train, nb_epoch=50)

    References
    ----------