cs230-code-examples icon indicating copy to clipboard operation
cs230-code-examples copied to clipboard

Published 20 hours ago • verified publisher icon cs230-stanford

Added example for prediction

Open pehpa opened this issue 6 years ago • 0 comments

Hello,

I just added an example to the cs230 code for simple prediction of a set of images. For this, I added the following files according to the train/training and evaluate/evaluation split:

tensorflow/vision/predict.py
tensorflow/vision/prediction.py

Additionally, I had to change the input_fn and model_fn functions to work not only for TRAIN and EVAL but also for the PREDICT case.

I think this change is quite helpful since I myself have searched a lot for examples on how to do the prediction with a saved TensorFlow model.

Best regards, Alex

pehpa avatar May 03 '18 15:05 pehpa