edhenry / chexnet

Implementation and fullstack pipeline for CheXNet classifier
MIT License
13 stars 12 forks source link

SavedModel support #2

Closed edhenry closed 5 years ago

edhenry commented 5 years ago

TensorFlow has support for exporting models using SavedModel that can be deployed using TensorFlow Serving.

The Callback class within keras.models.Model can be extended to support this within this implementation.

Opening issue and will implement proper branching structure for this repo.

edhenry commented 5 years ago

SavedModel support has been added and integrated with the chexnet_client.py module. TensorFlow Serving now returns serializes the model to the appropriate ./experiments/ directory.