manikbhandari / ebay-task

Predict the decade of a review using its text.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ebay-task

Predict the decade of a review using its text.

Data preprocess

Use EDA.ipynb to analyse the data and split into train, val and test

Models

Use run_classical_models.py to run the classical ML models and bert_classification.py to run a BERT based model. See the args for a description of the options.

About

Predict the decade of a review using its text.

License:MIT License


Languages

Language:Jupyter Notebook 99.3%Language:Python 0.7%