We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
full
value
1 parent b68b4da commit 4bf69c0Copy full SHA for 4bf69c0
tensorflow_model_analysis/extractors/tflite_predict_extractor.py
@@ -111,9 +111,7 @@ def _batch_reducible_process(
111
value = None
112
if value is None or np.any(np.equal(value, None)):
113
default = -1 if input_type in [np.float32, np.int64] else ''
114
- value = np.empty(input_shape)
115
- value.fill(default)
116
- value = value.astype(input_type)
+ value = np.full(input_shape, default, dtype=input_type)
117
logging.log_every_n(logging.WARNING,
118
'Feature %s not found. Setting default value.',
119
100, input_name)
0 commit comments