Unverified Commit 354cb7c2 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #1298 from peastman/examples

Fixes to examples
parents 1eb4a291 9d6c9854
Loading
Loading
Loading
Loading
+3 −4
Original line number Diff line number Diff line
@@ -87,13 +87,14 @@ class BPSymmetryFunctionRegression(TensorGraph):
                        dataset,
                        epochs=1,
                        predict=False,
                        deterministic=True,
                        pad_batches=True):
    for epoch in range(epochs):
      if not predict:
        print('Starting epoch %i' % epoch)
      for (X_b, y_b, w_b, ids_b) in dataset.iterbatches(
          batch_size=self.batch_size,
          deterministic=True,
          deterministic=deterministic,
          pad_batches=pad_batches):

        feed_dict = dict()
@@ -296,9 +297,7 @@ class ANIRegression(TensorGraph):
        jac=self.grad_one,
        method="BFGS",
        tol=1e-6,
        options={
            'disp': True
        })
        options={'disp': True})

    return res.x.reshape((num_atoms, 3))

+1 −1
Original line number Diff line number Diff line
@@ -106,7 +106,7 @@ def load_chembl(shard_size=2000,
    for transformer in transformers:
      dataset = transformer.transform(dataset)

  if spit == None:
  if split == None:
    return chembl_tasks, (dataset, None, None), transformers

  splitters = {
+1 −1
Original line number Diff line number Diff line
@@ -7,7 +7,7 @@ from __future__ import unicode_literals

import numpy as np

from models import GraphConvModel
from deepchem.models import GraphConvModel

np.random.seed(123)
import tensorflow as tf
+1 −1
Original line number Diff line number Diff line
@@ -7,7 +7,7 @@ from __future__ import unicode_literals

import numpy as np

from models import GraphConvModel
from deepchem.models import GraphConvModel

np.random.seed(123)
import tensorflow as tf
+1 −1
Original line number Diff line number Diff line
@@ -10,7 +10,7 @@ np.random.seed(123)
import tensorflow as tf
tf.set_random_seed(123)
import deepchem as dc
from membrain_permeability_datasets import load_permeability
from membrane_permeability_datasets import load_permeability

# Load Tox21 dataset
permeability_tasks, permeability_datasets, transformers = load_permeability(
Loading