vpj / bert-new-classification-sample

Trying out MLFlow and TorchServe

Train

Runs news_classifier.py

make train

Torch-serve

Start

make torchserve-start

Stop

make torchserve-stop

Deploy the trained model

make deploy

Get a sample prediction

make predict