model_training.ipynb
Python 3.10 • Last saved 2m ago
1. Train Neural Network Model
We initialize the sequential model and fit it to our processed dataset over 5 epochs.
In [12]
import tensorflow as tf
model = tf.keras.models.Sequential([
tf.keras.layers.Dense(128, activation='relu'),
tf.keras.layers.Dropout(0.2),
tf.keras.layers.Dense(10)
])
history = model.fit(x_train, y_train, epochs=5)