Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from tensorflow import keras
- import tensorflow as tf
- from keras.models import Sequential
- from keras.layers import Dense
- classifier = keras.models.Sequential()
- classifier.add(tf.layers.Dense(units = 6, kernel_initializer = keras.initializers.he_uniform(), activation = tf.nn.relu, input_shape =(11,)))
- classifier.add(tf.layers.Dense(units = 6, kernel_initializer = keras.initializers.he_uniform(), activation = tf.nn.relu))
- classifier.add(tf.layers.Dense(units = 1, kernel_initializer = tf.keras.initializers.he_uniform(), activation = tf.nn.softmax))
- classifier.compile(optimizer=tf.keras.optimizers.Adam(lr=0.0001),
- loss=tf.keras.losses.binary_crossentropy,
- metric=tf.keras.metrics.categorical_accuracy)
- my_estimator = tf.keras.estimator.model_to_estimator(keras_model=classifier)
Add Comment
Please, Sign In to add comment