This is targeting CTL users running an evaluator job, as opposed to alternating train/evaluate: 1) reading checkpoints as one becomes available, 2) evaluating the model with the dataset set aside for evaluation, 3) logging the evaluated metric, optionally generating summary file for TensorBoard, and optionally exporting the model considered the best so far for serving (TODO). It is possible for users to create such custom loop for evaluation, but this utility may be useful for standard, continuous evaluation that takes care of some details. This is also for feature parity with estimator. PiperOrigin-RevId: 339623940 Change-Id: I057fd255d46543003217ebbdc339155717c06044
41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
# Copyright 2015 The TensorFlow Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ==============================================================================
|
|
"""Implementation of the Keras API meant to be a high-level API for TensorFlow.
|
|
|
|
Detailed documentation and user guides are available at
|
|
[tensorflow.org](https://www.tensorflow.org/guide/keras).
|
|
"""
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
|
|
# pylint: disable=unused-import
|
|
from tensorflow.python import tf2
|
|
from tensorflow.python.keras import distribute
|
|
|
|
# See b/110718070#comment18 for more details about this import.
|
|
from tensorflow.python.keras import models
|
|
|
|
|
|
from tensorflow.python.keras.engine.input_layer import Input
|
|
from tensorflow.python.keras.engine.sequential import Sequential
|
|
from tensorflow.python.keras.engine.training import Model
|
|
|
|
from tensorflow.python.util.tf_export import keras_export
|
|
|
|
__version__ = '2.5.0'
|
|
|
|
keras_export('keras.__version__').export_constant(__name__, '__version__')
|