(flags_obj)
| 235 | |
| 236 | |
| 237 | def import_and_eval(flags_obj): |
| 238 | export_path = os.path.join(flags_obj.model_dir, 'export') |
| 239 | model = tf.saved_model.restore(export_path) |
| 240 | _, (x_test, y_test) = tf.keras.datasets.mnist.load_data() |
| 241 | x_test = x_test / np.float32(255) |
| 242 | y_predict = model(x_test) |
| 243 | accuracy = compute_accuracy(y_predict, y_test) |
| 244 | print('Model accuracy: {:0.2f}%'.format(accuracy.numpy() * 100)) |
| 245 | |
| 246 | |
| 247 | def apply_clean(flags_obj): |