Commit d695f5b3 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

fix

parent aa9e7640

deepchem/models/layers.py

+1 −0
Original line number Diff line number Diff line
@@ -2267,6 +2267,7 @@ class WeaveGather(tf.keras.layers.Layer): 
    return output_molecules



  def gaussian_histogram(self, x):

    import tensorflow_probability as tfp

    gaussian_memberships = [(-1.645, 0.283), (-1.080, 0.170), (-0.739, 0.134),

                            (-0.468, 0.118), (-0.228, 0.114), (0., 0.114),

                            (0.228, 0.114), (0.468, 0.118), (0.739, 0.134),

deepchem/rl/a2c.py

+1 −0
Original line number Diff line number Diff line
@@ -60,6 +60,7 @@ class A2CLossContinuous(object): 
    self.value_index = value_index



  def __call__(self, outputs, labels, weights):

    import tensorflow_probability as tfp

    mean = outputs[self.mean_index]

    std = outputs[self.std_index]

    value = outputs[self.value_index]

CRA Git | Maintained and supported by SUSTech CRA and CCSE