Skip to content

Instantly share code, notes, and snippets.

@jcblw

jcblw/create_model_train.py

Created October 10, 2022 15:50
Show Gist options
  • Save jcblw/e26f98aa247c1aa1035f85afad7c683c to your computer and use it in GitHub Desktop.
Save jcblw/e26f98aa247c1aa1035f85afad7c683c to your computer and use it in GitHub Desktop.
Download ZIP
Used for a Medium article
Raw
create_model_train.py
# Create a model for predicting the next word
model = tf.keras.Sequential()
# Load the data from local CSV file
data = pd.read_csv('data.csv')
# Split the data into training and testing
train, test = train_test_split(data, test_size=0.2)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment