https://github.com/jmisilo/amazon-reviews-rating
https://github.com/jmisilo/amazon-reviews-rating
Last synced: about 1 year ago
JSON representation
- Host: GitHub
- URL: https://github.com/jmisilo/amazon-reviews-rating
- Owner: jmisilo
- Created: 2022-02-05T22:01:59.000Z (over 4 years ago)
- Default Branch: main
- Last Pushed: 2022-02-15T20:49:21.000Z (over 4 years ago)
- Last Synced: 2025-02-09T20:36:09.386Z (over 1 year ago)
- Language: Jupyter Notebook
- Size: 83 KB
- Stars: 0
- Watchers: 1
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: readme.md
Awesome Lists containing this project
README
# Prediction star rating of game reviews
### Description
Project created for predicting rating based on game review. It's goal is to improve my skills of training nlp models and learn new architecture - transformers.
### Model
Model based on transformers architecture, that are built with tutorial created by [Aladdin Persson](https://github.com/aladdinpersson). Model contains 3 transformers that are preceded by word-level and position-level embedding layers. After tranformers there are 3 fully connected layers. Model trained using [kaggle.com](kaggle.com) GPUs. Training took ~6 hours.
### Scores
Model achieved ~0.8 F1-score and ~0.87 ROC AUC score. Accuracy is not so important, beacouse even after augmentation data was quite imbalanced.