Skip to content

Instantly share code, notes, and snippets.

@shayaf84

shayaf84/bertCompile.py

Created February 20, 2022 23:00
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 shayaf84/0593b2e15de2e7f2c5d1f50bbf3ffcb1 to your computer and use it in GitHub Desktop.
Save shayaf84/0593b2e15de2e7f2c5d1f50bbf3ffcb1 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
bertCompile.py
#Compile model on adam optimizer, binary_crossentropy loss, and accuracy metrics
model.compile(optimizer='adam',loss='binary_crossentropy',metrics=['accuracy'])
#Train model on 5 epochs
model.fit(title_train,y_train,epochs=5)
#Evaluate model on test data
model.evaluate(title_test,y_test)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment