Skip to content

Instantly share code, notes, and snippets.

@techwithshadab

techwithshadab/Tensorflow for Beginners(Add 3 Dense Layers).py

Created August 2, 2019 18:46
Show Gist options
  • Save techwithshadab/17331c482208232ff83eb0a3a88f7cbc to your computer and use it in GitHub Desktop.
Save techwithshadab/17331c482208232ff83eb0a3a88f7cbc to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Tensorflow for Beginners(Add 3 Dense Layers).py
l_0 = tf.keras.layers.Dense(units=4, input_shape=[1])
l_1 = tf.keras.layers.Dense(units=5)
l_2 = tf.keras.layers.Dense(units=1)
model = tf.keras.Sequential([l_0, l_1, l_2])
model.compile(loss='mean_squared_error', optimizer=tf.keras.optimizers.Adam(0.1))
model.fit(X_train,y_train, epochs=2000,verbose=False)
print('\n Finished training Model')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment