def build_network()

in recommended-item-search/softmax_model.py [0:0]

• 10 lines of code
• 2 McCabe index (conditional complexity)

def build_network(
    inputs, hidden_dims, activation_fn, scope='user_embeddings'):
  """Build a forward network graph."""
  with tf.variable_scope(scope):
    input_dim = inputs.shape[1].value
    for i, output_dim in enumerate(hidden_dims):
      layer = tf.layers.Dense(output_dim, activation=activation_fn)
      outputs = layer(inputs)
      input_dim = output_dim
      inputs = outputs
    return outputs