Commit c773e34b authored by Karl Leswing's avatar Karl Leswing
Browse files

Correct attr

parent 9d7bc0e1

deepchem/dock/tests/test_docking.py

+1 −1
Original line number Diff line number Diff line
@@ -11,7 +11,7 @@ __license__ = "MIT" 


import unittest

import os

from nose.tools import attr

from nose.plugins.attrib import attr

import sys

import deepchem as dc

deepchem/models/tensorgraph/tests/test_tensor_graph.py

+1 −1
Original line number Diff line number Diff line
@@ -151,7 +151,7 @@ class TestTensorGraph(unittest.TestCase): 
    label = Label(shape=(None, 2))

    smce = SoftMaxCrossEntropy(in_layers=[label, dense])

    loss = ReduceMean(in_layers=[smce])

    tg = dc.models.TensorGraph(learning_rate=1.0, use_queue=False)

    tg = dc.models.TensorGraph(learning_rate=0.1, use_queue=False)

    tg.add_output(output)

    tg.set_loss(loss)

    tg.fit(dataset, nb_epoch=1000)

CRA Git | Maintained and supported by SUSTech CRA and CCSE