/DiabetesDetection

Primary LanguageJupyter Notebook

Usage

This trains a Xgboost model on Pima Indians Diabetes Database

The logs are stored at logs.txt

The commands are stored at commands.txt

Docker image of the triton api is stored at susnato/xgb-diabetes-triton:cpu To run-

curl -X POST http://localhost:8000/v2/models/xgb/infer -d @/home/susnato/PycharmProjects/DiabetesDetection/triton_payload.json

where the payload should look like -

{"inputs": [{"name": "input__0", "datatype": "FP32", "shape": [1,8], "data": [[1,85,66,29,0,26.6,0.351,31]]}]}

The output should be -

{"model_name":"xgb","model_version":"1","outputs":[{"name":"output__0","datatype":"FP32","shape":[1,2],"data":[0.6297283172607422,0.3702717125415802]}]}