INNER CODE UNIT · Python

get_num_tasks

deepchem/deepchem · contrib/tensorflow_models/__init__.py:228

  def get_num_tasks(self):
    return self.n_tasks

  def construct_graph(self, training, seed):
    """Returns a TensorflowGraph object."""
    graph = tf.Graph()

    # Lazily created by _get_shared_session().
    shared_session = None

    # Cache of TensorFlow scopes, to prevent '_1' appended scope names
    # when subclass-overridden methods use the same scopes.
    name_scopes = {}

    # Setup graph
    with graph.as_default():
      if seed is not None:
        tf.set_random_seed(seed)

View source record →