Loading deepchem/models/layers.py +4 −4 Original line number Diff line number Diff line Loading @@ -391,7 +391,7 @@ class GraphConvolutionLayer(tf.keras.layers.Layer): def __init__(self, units: int, activation: Callable = "tanh", activation: Callable = activations.tanh, dropout_rate: float = 0.0, edges: int = 5, name: str = "", Loading Loading @@ -491,7 +491,7 @@ class GraphAggregationLayer(tf.keras.layers.Layer): def __init__(self, units: int = 128, activation: Callable = "tanh", activation: Callable = activations.tanh, dropout_rate: float = 0.0, name: str = "", **kwargs): Loading Loading @@ -570,7 +570,7 @@ class MultiGraphConvolutionLayer(tf.keras.layers.Layer): def __init__(self, units: Tuple = (128, 64), activation: Callable = "tanh", activation: Callable = activations.tanh, dropout_rate: float = 0.0, edges: int = 5, name: str = "", Loading Loading @@ -664,7 +664,7 @@ class GraphEncoderLayer(tf.keras.layers.Layer): def __init__(self, units: List = [(128, 64), 128], activation: Callable = "tanh", activation: Callable = activations.tanh, dropout_rate: float = 0.0, edges: int = 5, name: str = "", Loading deepchem/models/tests/test_molgan_layers.py +5 −4 Original line number Diff line number Diff line Loading @@ -2,6 +2,7 @@ import unittest from tensorflow import keras from tensorflow.keras.layers import Input from tensorflow.keras import activations from deepchem.models.layers import GraphConvolutionLayer, MultiGraphConvolutionLayer, GraphAggregationLayer, GraphEncoderLayer Loading @@ -27,7 +28,7 @@ class test_molgan_layers(unittest.TestCase): (None, vertices, nodes), (None, vertices, units))] assert layer.units == units assert layer.activation == 'tanh' assert layer.activation == activations.tanh assert layer.edges == 5 assert layer.dropout_rate == 0.0 Loading @@ -42,7 +43,7 @@ class test_molgan_layers(unittest.TestCase): assert model.output_shape == (None, units) assert layer.units == units assert layer.activation == 'tanh' assert layer.activation == activations.tanh assert layer.dropout_rate == 0.0 def test_multigraph_convolution_layer(self): Loading @@ -62,7 +63,7 @@ class test_molgan_layers(unittest.TestCase): assert model.output_shape == (None, vertices, second_convolution_unit) assert layer.units == units assert layer.activation == 'tanh' assert layer.activation == activations.tanh assert layer.edges == 5 assert layer.dropout_rate == 0.0 Loading @@ -87,7 +88,7 @@ class test_molgan_layers(unittest.TestCase): assert layer.graph_convolution_units == (first_convolution_unit, second_convolution_unit) assert layer.auxiliary_units == aggregation_unit assert layer.activation == 'tanh' assert layer.activation == activations.tanh assert layer.edges == 5 assert layer.dropout_rate == 0.0 Loading Loading
deepchem/models/layers.py +4 −4 Original line number Diff line number Diff line Loading @@ -391,7 +391,7 @@ class GraphConvolutionLayer(tf.keras.layers.Layer): def __init__(self, units: int, activation: Callable = "tanh", activation: Callable = activations.tanh, dropout_rate: float = 0.0, edges: int = 5, name: str = "", Loading Loading @@ -491,7 +491,7 @@ class GraphAggregationLayer(tf.keras.layers.Layer): def __init__(self, units: int = 128, activation: Callable = "tanh", activation: Callable = activations.tanh, dropout_rate: float = 0.0, name: str = "", **kwargs): Loading Loading @@ -570,7 +570,7 @@ class MultiGraphConvolutionLayer(tf.keras.layers.Layer): def __init__(self, units: Tuple = (128, 64), activation: Callable = "tanh", activation: Callable = activations.tanh, dropout_rate: float = 0.0, edges: int = 5, name: str = "", Loading Loading @@ -664,7 +664,7 @@ class GraphEncoderLayer(tf.keras.layers.Layer): def __init__(self, units: List = [(128, 64), 128], activation: Callable = "tanh", activation: Callable = activations.tanh, dropout_rate: float = 0.0, edges: int = 5, name: str = "", Loading
deepchem/models/tests/test_molgan_layers.py +5 −4 Original line number Diff line number Diff line Loading @@ -2,6 +2,7 @@ import unittest from tensorflow import keras from tensorflow.keras.layers import Input from tensorflow.keras import activations from deepchem.models.layers import GraphConvolutionLayer, MultiGraphConvolutionLayer, GraphAggregationLayer, GraphEncoderLayer Loading @@ -27,7 +28,7 @@ class test_molgan_layers(unittest.TestCase): (None, vertices, nodes), (None, vertices, units))] assert layer.units == units assert layer.activation == 'tanh' assert layer.activation == activations.tanh assert layer.edges == 5 assert layer.dropout_rate == 0.0 Loading @@ -42,7 +43,7 @@ class test_molgan_layers(unittest.TestCase): assert model.output_shape == (None, units) assert layer.units == units assert layer.activation == 'tanh' assert layer.activation == activations.tanh assert layer.dropout_rate == 0.0 def test_multigraph_convolution_layer(self): Loading @@ -62,7 +63,7 @@ class test_molgan_layers(unittest.TestCase): assert model.output_shape == (None, vertices, second_convolution_unit) assert layer.units == units assert layer.activation == 'tanh' assert layer.activation == activations.tanh assert layer.edges == 5 assert layer.dropout_rate == 0.0 Loading @@ -87,7 +88,7 @@ class test_molgan_layers(unittest.TestCase): assert layer.graph_convolution_units == (first_convolution_unit, second_convolution_unit) assert layer.auxiliary_units == aggregation_unit assert layer.activation == 'tanh' assert layer.activation == activations.tanh assert layer.edges == 5 assert layer.dropout_rate == 0.0 Loading
