Commit ae8be6cf authored by Milosz Grabski's avatar Milosz Grabski
Browse files

fixed example for MolGANEncoderLayer

parent 72652d8b
Loading
Loading
Loading
Loading
+28 −28
Original line number Diff line number Diff line
@@ -387,17 +387,17 @@ class MolGANConvolutionLayer(tf.keras.layers.Layer):
  --------
  See: MolGANMultiConvolutionLayer for using in layers.

  vertices = 9
  nodes = 5
  edges = 5
  units = 128

  layer = MolGANConvolutionLayer(units=units,edges=edges)
  adjacency_tensor= layers.Input(shape=(vertices, vertices, edges))
  node_tensor = layers.Input(shape=(vertices,nodes))
  hidden1 = layer([adjacency_tensor,node_tensor])
  output = layer(hidden1)
  model = keras.Model(inputs=[adjacency_tensor,node_tensor], outputs=[output])
  >>> vertices = 9
  >>> nodes = 5
  >>> edges = 5
  >>> units = 128

  >>> layer = MolGANConvolutionLayer(units=units,edges=edges)
  >>> adjacency_tensor= layers.Input(shape=(vertices, vertices, edges))
  >>> node_tensor = layers.Input(shape=(vertices,nodes))
  >>> hidden1 = layer([adjacency_tensor,node_tensor])
  >>> output = layer(hidden1)
  >>> model = keras.Model(inputs=[adjacency_tensor,node_tensor], outputs=[output])

  References
  ----------
@@ -723,7 +723,7 @@ class MolGANEncoderLayer(tf.keras.layers.Layer):

  Example
  --------
  >>> def create_discriminator(adjacency_tensor, node_tensor):
  
  vertices = 9
  edges = 5
  dropout_rate = .0
@@ -739,7 +739,7 @@ class MolGANEncoderLayer(tf.keras.layers.Layer):
  dense = layers.Dropout(dropout_rate)(dense)
  output = layers.Dense(units=1)(dense)

        return keras.Model(inputs=[adjacency_tensor,node_tensor], outputs=[output])
  keras.Model(inputs=[adjacency_tensor,node_tensor], outputs=[output])

  References
  ----------