Commit 032ea520 authored by leswing's avatar leswing
Browse files

Test fixes

parent 54e8b170
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
import numpy as np
from data import NumpyDataset
from nose.tools import assert_true
from unittest import TestCase

import deepchem as dc
from data.datasets import Databag
from deepchem.data import NumpyDataset
from deepchem.data.datasets import Databag
from deepchem.models.tensorgraph.layers import Dense, ReduceMean, SoftMax, SoftMaxCrossEntropy
from deepchem.models.tensorgraph.layers import Feature, Label
from deepchem.models.tensorgraph.layers import ReduceSquareDifference
+2 −6
Original line number Diff line number Diff line
@@ -50,8 +50,7 @@ def graph_conv_model(batch_size, tasks):
  batch_norm2 = BatchNormLayer(in_layers=[gc2])
  gp2 = GraphPoolLayer(
      in_layers=[batch_norm2, degree_slice, membership] + deg_adjs)
  dense = Dense(
      out_channels=128, activation_fn=None, in_layers=[gp2])
  dense = Dense(out_channels=128, activation_fn=None, in_layers=[gp2])
  batch_norm3 = BatchNormLayer(in_layers=[dense])
  gg1 = GraphGather(
      batch_size=batch_size,
@@ -61,10 +60,7 @@ def graph_conv_model(batch_size, tasks):
  costs = []
  labels = []
  for task in tasks:
    classification = Dense(
        out_channels=2,
        activation_fn=None,
        in_layers=[gg1])
    classification = Dense(out_channels=2, activation_fn=None, in_layers=[gg1])

    softmax = SoftMax(in_layers=[classification])
    model.add_output(softmax)