Skip to content

Instantly share code, notes, and snippets.

@zabop
Created July 17, 2021 21:09
Show Gist options
  • Save zabop/4c34616435b52bdd021d7b67d7c3ce07 to your computer and use it in GitHub Desktop.
Save zabop/4c34616435b52bdd021d7b67d7c3ce07 to your computer and use it in GitHub Desktop.
model1 = models.Sequential()
model1.add(layers.Dense(16, activation='relu', input_shape=(10000,)))
model1.add(layers.Dense(16, activation='relu'))
model1.add(layers.Dense(1, activation='sigmoid'))
model1.compile(optimizer='rmsprop',
loss='binary_crossentropy',
metrics=['acc'])
history1 = model1.fit(partial_x_train,
partial_y_train,
epochs=epochs,
batch_size=512,
validation_data=(x_val, y_val))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment