def make_keras_model(initializer='ones'):
    model_input = keras.Input(shape=(10,))
    x = keras.layers.Dense(64, kernel_initializer=initializer, bias_initializer='zeros')(model_input)
    x = keras.layers.Dense(128, kernel_initializer=initializer, bias_initializer='zeros')(x)
    x = keras.layers.Dense(256, kernel_initializer=initializer, bias_initializer='zeros')(x)
    x = keras.layers.Dense(256, kernel_initializer=initializer, bias_initializer='zeros')(x)
    x = keras.layers.Dense(10, kernel_initializer=initializer, bias_initializer='zeros')(x)
    return keras.Model(inputs=model_input, outputs=x)
def make_keras_model(initializer='ones'):
    model_input = keras.Input(shape=(10,))
    x = keras.layers.Dense(64, kernel_initializer=initializer, bias_initializer='zeros')(model_input)
    x = keras.layers.Dense(128, kernel_initializer=initializer, bias_initializer='zeros')(x)
    x = keras.layers.Dense(256, kernel_initializer=initializer, bias_initializer='zeros')(x)
    x = keras.layers.Dense(256, kernel_initializer=initializer, bias_initializer='zeros')(x)
    x = keras.layers.Dense(10, kernel_initializer=initializer, bias_initializer='zeros')(x)
    return keras.Model(inputs=model_input, outputs=x)