Commit b631fd3a authored by ZHENQIN WU's avatar ZHENQIN WU
Browse files

fix bugs

parent 86a93af4

deepchem/models/tensorgraph/graph_layers.py

+1 −1
Original line number Diff line number Diff line
@@ -309,7 +309,7 @@ class WeaveGather(Layer): 
                            (0.228, 0.114), (0.468, 0.118), (0.739, 0.134),

                            (1.080, 0.170), (1.645, 0.283)]

    dist = [

        tf.contrib.distributions.Normal(loc=p[0], scale=p[1])

        tf.contrib.distributions.Normal(p[0], p[1])

        for p in gaussian_memberships

    ]

    dist_max = [dist[i].prob(gaussian_memberships[i][0]) for i in range(11)]

deepchem/nn/weave_layers.py

+1 −1
Original line number Diff line number Diff line
@@ -413,7 +413,7 @@ class WeaveGather(Layer): 
                            (0.228, 0.114), (0.468, 0.118), (0.739, 0.134),

                            (1.080, 0.170), (1.645, 0.283)]

    dist = [

        tf.contrib.distributions.Normal(loc=p[0], scale=p[1])

        tf.contrib.distributions.Normal(p[0], p[1])

        for p in gaussian_memberships

    ]

    dist_max = [dist[i].prob(gaussian_memberships[i][0]) for i in range(11)]

CRA Git | Maintained and supported by SUSTech CRA and CCSE