Skip to content
This repository has been archived by the owner on Dec 31, 2020. It is now read-only.

Drew/mnist dense #17

Open
wants to merge 6 commits into
base: master
Choose a base branch
from

Conversation

DrewParallel
Copy link
Contributor

Add saved model for MNIST_dense that includes a prediction signature.
Create a PredictionMethod to use the prediction signature.
Create a test for the saved model and prediction signature.

DrewParallel added 6 commits May 26, 2017 18:51
* protobuf: remove protobuf files for tensorflow framework and take a dependency on the jar instead.

* Remove xolstice from pom since we no longer need to generate protobufs
This model represents a dense model for MNIST data.
It has standard prediction signature for tf serve.
Add a method for pulling the prediction method from the saved model signature.
This model defines the predict function to call in the
prediction signature of the MNIST_dense saved model.
This test uses the first 5 values of the MNIST test set to verify that
the MNIST model prediction works as expected.
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant