Skip to content

Instantly share code, notes, and snippets.

@frogermcs
Created May 27, 2019 19:33
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save frogermcs/4210f8f50a5c79324c812b2020e54309 to your computer and use it in GitHub Desktop.
Save frogermcs/4210f8f50a5c79324c812b2020e54309 to your computer and use it in GitHub Desktop.
model = tf.keras.Sequential([
hub.KerasLayer("https://tfhub.dev/google/tf2-preview/mobilenet_v2/feature_vector/4",
output_shape=[1280],
trainable=False),
tf.keras.layers.Dropout(0.4),
tf.keras.layers.Dense(train_generator.num_classes, activation='softmax')
])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment