https://github.com/pegasystems/nlp-model-containers
https://github.com/pegasystems/nlp-model-containers
Last synced: 12 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/pegasystems/nlp-model-containers
- Owner: pegasystems
- License: apache-2.0
- Created: 2020-06-17T19:18:32.000Z (about 6 years ago)
- Default Branch: master
- Last Pushed: 2024-03-29T18:00:06.000Z (over 2 years ago)
- Last Synced: 2025-05-12T19:54:19.522Z (about 1 year ago)
- Language: Jupyter Notebook
- Size: 12.3 MB
- Stars: 2
- Watchers: 4
- Forks: 3
- Open Issues: 3
-
Metadata Files:
- Readme: README.md
- License: LICENSE
- Security: SECURITY.md
Awesome Lists containing this project
README
## Table of contents
* [General info](#general-info)
* [Machine Learning NLP container](#machine-learning-nlp-container)
* [Deep Learning NLP container](#deep-learning-nlp-container)
## General info
This is an example dockerized container which comes with the dependencies required to run your machine learning and deep learning models
This uses flask to serve the models using a rest end point secured with Authlib for OAuth2 (using https://docs.authlib.org/en/latest/flask/2/) as well as
an end point without auth.
Notes: Please note that it is a sample Docker Container. scripts can be modified and deployed with/without authentication.
The Repo contains 2 containers.
* Machine Learning Container
* Deep Learning Container
Please find the steps to be followed:
* Step1: Train the model
* Step2: Save the model
* Step3: Dump the saved model into specific location
* Step4: Initialize the Flask API via Docker container
* Step5: Predict and access the endpoints
Note: Detailed steps are available in the Machine Learning and Deep Learning respective README.md files
Machine Learning:
https://github.com/pegasystems/nlp-model-containers/tree/master/machine-learning-nlp-container
Deep Learning:
https://github.com/pegasystems/nlp-model-containers/tree/master/deep-learning-nlp-container
## Machine Learning NLP container
The project contains:
* README.md
* Dockerfile
* requirments.txt
* app.py
* sample training scripts(https://github.com/pegasystems/nlp-model-containers/tree/master/machine-learning-nlp-container/samples)
* sample saved models(https://github.com/pegasystems/nlp-model-containers/tree/master/machine-learning-nlp-container/models)
Algorithm Supported(Scikit/SKlearn):
* Boosting Algos( XGBoost/LightBoost/AdaBoost/CatBoost)
* Decision Tree
* Random Forest
* Naive Bayes(Multionomial and Gausian)
* SVM
* KNN
## Deep Learning NLP container
The project contains:
* README.md
* Dockerfile
* requirments.txt
* app.py
* sample training scripts(https://github.com/pegasystems/nlp-model-containers/tree/master/deep-learning-nlp-container/samples)
* sample saved models(https://github.com/pegasystems/nlp-model-containers/tree/master/deep-learning-nlp-container/models/keras_small_talk_model)
Algorithm Supported(Keras Framework):
* Tensorflow
* Microsoft Cognitive Toolkit (CNTK)
* Theano