Commit 1636ede4 authored by Peter Eastman's avatar Peter Eastman
Browse files

Bug fixes

parent 4f05c690

deepchem/models/IRV.py

+2 −2
Original line number Diff line number Diff line
@@ -118,8 +118,8 @@ class TensorflowMultitaskIRVClassifier(KerasModel): 
    outputs = layers.Stack(axis=1)(outputs)

    outputs2 = Lambda(lambda x: 1 - x)(outputs)

    outputs = [

        Concatenate(axis=2)([outputs2, outputs]),

        Concatenate(axis=1)(logits)

        Concatenate(axis=2)([outputs2, outputs]), logits[0]

        if len(logits) == 1 else Concatenate(axis=1)(logits)

    ]

    model = tf.keras.Model(inputs=[mol_features], outputs=outputs)

    super(TensorflowMultitaskIRVClassifier, self).__init__(

examples/clintox/clintox_graph_conv.py

+1 −1
Original line number Diff line number Diff line
@@ -8,7 +8,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

CRA Git | Maintained and supported by SUSTech CRA and CCSE