Commit 4e63fa40 authored by miaecle's avatar miaecle
Browse files

fix bugs and yapf

parent 771bad01
Loading
Loading
Loading
Loading
+0 −1
Original line number Diff line number Diff line
@@ -6,7 +6,6 @@ import numpy as np
import tensorflow as tf

from deepchem.utils.save import log
<<<<<<< HEAD
from deepchem.models.tensorgraph.tensor_graph import TensorGraph
from deepchem.models.tensorgraph.layers import Layer, SigmoidCrossEntropy, \
    Sigmoid, Feature, Label, Weights, Concat, WeightedError
+4 −3
Original line number Diff line number Diff line
@@ -5,7 +5,7 @@ from deepchem.models import TensorGraph
from deepchem.models.tensorgraph.graph_layers import WeaveLayer, WeaveGather, DTNNEmbedding, DTNNGather, DTNNStep, \
  DTNNExtract, DAGLayer, DAGGather, MessagePassing, SetGather
from deepchem.models.tensorgraph.layers import Feature, Conv1D, Dense, Flatten, Reshape, Squeeze, Transpose, \
  CombineMeanStd, Repeat, Gather, GRU, LSTM, L2Loss, Concat, SoftMax, Sigmoid, SigmoidCrossEntropy\
  CombineMeanStd, Repeat, Gather, GRU, LSTM, L2Loss, Concat, SoftMax, Sigmoid, SigmoidCrossEntropy, \
  Constant, Variable, StopGradient, Add, Multiply, Log, Exp, InteratomicL2Distances, \
  SoftMaxCrossEntropy, ReduceMean, ToFloat, ReduceSquareDifference, Conv2D, MaxPool2D, ReduceSum, GraphConv, GraphPool, \
  GraphGather, BatchNorm, WeightedError, \
@@ -154,6 +154,7 @@ def test_Softmax_pickle():
  tg.build()
  tg.save()


def test_Sigmoid_pickle():
  tg = TensorGraph()
  feature = Feature(shape=(tg.batch_size, 1))
@@ -163,6 +164,7 @@ def test_Sigmoid_pickle():
  tg.build()
  tg.save()


def test_Concat_pickle():
  tg = TensorGraph()
  feature = Feature(shape=(tg.batch_size, 1))
@@ -256,6 +258,7 @@ def test_SoftmaxCrossEntropy_pickle():
  tg.build()
  tg.save()


def test_SigmoidCrossEntropy_pickle():
  tg = TensorGraph()
  feature = Feature(shape=(tg.batch_size, 1))
@@ -669,5 +672,3 @@ def test_Slice_pickle():
  tg.set_loss(out)
  tg.build()
  tg.save()

+0 −1
Original line number Diff line number Diff line
@@ -457,7 +457,6 @@ class TestOverfit(test_util.TensorFlowTestCase):
    scores = model.evaluate(dataset_trans, [classification_metric])
    assert scores[classification_metric.name] > .9


  def test_sklearn_multitask_regression_overfit(self):
    """Test SKLearn singletask-to-multitask overfits tiny regression data."""
    n_tasks = 2