Skip to content

FastAPI wrapper to train a GPT-2 model using the IMDb dataset and to classify texts as positive or negative with 83% accuracy.

License

Notifications You must be signed in to change notification settings

gabrielganzer/GPT2-TextClassification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GPT2-TextClassification

About

FastAPI wrapper to train a GPT-2 model using the IMDb dataset and to classify texts as positive or negative with 83% accuracy.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages