- Make model with data text review cloth with transfer learning distilbert/distilbert-base-uncased
- Make a API with FastAPI
- Hyperparameter Reference using with this paper Journal Distilber and this Journal bert
- 0 Negative review
- 1 Neutral review
- 2 Positive review
http://127.0.0.1:8000/prediction/text?query={your text}
{
"data": {
"input_text": "i love pretty soft shirt falls well want wear everyday",
"result": "Positve"
}
}
@article{Sanh2019DistilBERTAD,
title={DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lighter},
author={Victor Sanh and Lysandre Debut and Julien Chaumond and Thomas Wolf},
journal={ArXiv},
year={2019},
volume={abs/1910.01108}
}