MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / make_keras_model

Function make_keras_model

tensorflow/python/eager/benchmarks_test.py:110–122  ·  view source on GitHub ↗
(initializer="ones")

Source from the content-addressed store, hash-verified

108
109
110def make_keras_model(initializer="ones"):
111 model_input = keras.Input(shape=(10,))
112 x = keras.layers.Dense(
113 64, kernel_initializer=initializer, bias_initializer="zeros")(model_input)
114 x = keras.layers.Dense(
115 128, kernel_initializer=initializer, bias_initializer="zeros")(x)
116 x = keras.layers.Dense(
117 256, kernel_initializer=initializer, bias_initializer="zeros")(x)
118 x = keras.layers.Dense(
119 256, kernel_initializer=initializer, bias_initializer="zeros")(x)
120 x = keras.layers.Dense(
121 10, kernel_initializer=initializer, bias_initializer="zeros")(x)
122 return keras.Model(inputs=model_input, outputs=x)
123
124
125def make_sequential_keras_model(initializer="ones"):

Calls 2

InputMethod · 0.45
ModelMethod · 0.45

Tested by

no test coverage detected