Commit 8147b09f authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

docs

parent 7ee86ae3
Loading
Loading
Loading
Loading
+1 −3
Original line number Diff line number Diff line
@@ -18,6 +18,4 @@ alternate modes of splitting. Similarly, since the source data
came pre-featurized, it is not possible to apply alternative
featurizations.

In this example, we train various models on the Factors dataset:

- 
In this example, we train various models on the Factors dataset.
+1 −3
Original line number Diff line number Diff line
@@ -18,6 +18,4 @@ are used:
References:
AIDS Antiviral Screen Data. https://wiki.nci.nih.gov/display/NCIDTPdata/AIDS+Antiviral+Screen+Data

## Models Trained

In this example we train the following models on the HIV collection.
In this example we train models on the HIV collection.
+1 −1
Original line number Diff line number Diff line
@@ -18,4 +18,4 @@ alternate modes of splitting. Similarly, since the source data
came pre-featurized, it is not possible to apply alternative
featurizations.

This folder contains examples training models on the Kaggle dataset:
This folder contains examples training models on the Kaggle dataset.
+1 −0
Original line number Diff line number Diff line
model/
+3 −5
Original line number Diff line number Diff line
@@ -2,15 +2,13 @@ import deepchem as dc
import rdkit.Chem as Chem

model = dc.models.ChemCeption(
    img_spec="engd",
    n_tasks=1,
    model_dir="./model",
    mode="regression")
    img_spec="engd", n_tasks=1, model_dir="./model", mode="regression")
model.restore()

smiles = "CCCCC"
featurizer = dc.feat.SmilesToImage(img_spec="engd", img_size=80, res=0.5)
dataset = dc.data.NumpyDataset(featurizer.featurize([Chem.MolFromSmiles(smiles)]))
dataset = dc.data.NumpyDataset(
    featurizer.featurize([Chem.MolFromSmiles(smiles)]))
prediction = model.predict(dataset)
print("smiles: %s" % smiles)
print("prediction: %s" % str(prediction))
+1 −1

File changed.

Contains only whitespace changes.

Loading