Skip to content
Snippets Groups Projects
Commit b8854cb2 authored by Riccardo Riva's avatar Riccardo Riva Committed by Mads M. Pedersen
Browse files

Added batch_size option to TensorFlow predict. Should improve performance

parent 23057e1d
No related branches found
No related tags found
No related merge requests found
......@@ -101,7 +101,7 @@ class TensorflowSurrogate():
mi, ma = self.input_scaler.data_min_[i], self.input_scaler.data_max_[i]
warnings.warn(f"Input, {k}, with value, {max_v} outside range {mi}-{ma}")
return self.output_scaler.inverse_transform(self.model.predict(x_scaled))
return self.output_scaler.inverse_transform(self.model.predict(x_scaled, batch_size=x.shape[0]))
@property
def input_space(self):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment