/access-niu

DIY NIU

Primary LanguagePythonApache License 2.0Apache-2.0

Build Status Python 3.6 License Code style: black

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.