Skip to content

Using pretrained BERT to perform inference for sentence classification.

Notifications You must be signed in to change notification settings

jyu-theartofml/BERT_seq_classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

BERT_seq_classification

Using pretrained BERT to perform inference for sentence classification.

The notebook in this repo presents steps to generate label predictions from sentence inputs in Pytorch. This was done after training a BERT model according to the tutorial by huggingface and the model is saved in a directory. The huggingface repo does not currently have a predict module or Class but it does offer an option for evaluation as the model is being trained.

About

Using pretrained BERT to perform inference for sentence classification.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published