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

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