Skip to content

Instantly share code, notes, and snippets.

@Wapiti08

Wapiti08/bert_basic.py Secret

Created June 21, 2022 09:20
Show Gist options
  • Save Wapiti08/864bf134106d59ccc57f2f7eda045910 to your computer and use it in GitHub Desktop.
Save Wapiti08/864bf134106d59ccc57f2f7eda045910 to your computer and use it in GitHub Desktop.
Download ZIP
bert_basic
Raw
bert_basic.py
bert_model = TFBertForSequenceClassification.from_pretrained(model_name, num_labels=num_classes)
bert_model.compile(loss='categorical_crossentropy', optimizer='adam',metrics=['accuracy'])
bert_model.summary()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment