Searched refs:EstimatorModeKeys (Results 1 – 4 of 4) sorted by relevance
41 class EstimatorModeKeys(object): class57 return mode in [KerasModeKeys.PREDICT, EstimatorModeKeys.PREDICT]61 return mode in [KerasModeKeys.TEST, EstimatorModeKeys.EVAL]65 return mode in [KerasModeKeys.TRAIN, EstimatorModeKeys.TRAIN]
35 self.assertEqual(3, mode_map[mode_keys.EstimatorModeKeys.PREDICT])37 self.assertEqual(1, mode_map[mode_keys.EstimatorModeKeys.EVAL])41 _ = mode_map[mode_keys.EstimatorModeKeys.TRAIN]60 mode_keys.EstimatorModeKeys.PREDICT: 1
3 is_instance: "<class \'tensorflow.python.saved_model.model_utils.mode_keys.EstimatorModeKeys\'>"