Function _get_model

tensorflow/python/keras/optimizers_test.py:35–41

(input_dim, num_hidden, output_dim)

def _get_model(input_dim, num_hidden, output_dim):
  model = keras.models.Sequential()
  model.add(keras.layers.Dense(num_hidden,
                               activation='relu',
                               input_shape=(input_dim,)))
  model.add(keras.layers.Dense(output_dim, activation='softmax'))
  return model


@keras_parameterized.run_all_keras_modes