Prediction Service

Machine Learning model training and prediction api for brand prediction.

Train Model

Set Up Project

virtualenv venv
source venv/bin/activate
pip install -r requirements.txt

Train and Deploy models to S3

python src/model/train.py

Deploy Predicton Api

Set Up Serverless

npm install -g serverless
npm i -D serverless-dotenv-plugin
sls plugin install -n serverless-python-requirements

Deploy prediction Api to AWS lambda

sls deploy