akash9182 / access-niu

DIY NIU

access-niu

This repository contains application to train models for Image classification and Regression tasks.

Tasks

• Create a basic app for training and inference
• Support for training regression models
• Support of training multi input/output models
• Incorporate Bayesian Inference for finding uncertainty in the predictions.
• Create Docker Image
• Support for serving the application with gunicorn
• Anything else?

Installation

pip install access-niu

Training

python -m access_niu.train --template access_niu/sample/colors/sample_template.yml

Inference

python -m access_niu.wsgi --project ./output/colors

Now use this curl command to parse

curl -X POST \
  http://localhost:8000/parse \
  -F data=@test_image.jpg

References

• This project is inspired from RASA-NLU project.