From b8854cb21129959a017c19f11438719863a2d5cc Mon Sep 17 00:00:00 2001
From: Riccardo Riva <ricriv@dtu.dk>
Date: Wed, 13 Oct 2021 08:07:52 +0200
Subject: [PATCH] Added batch_size option to TensorFlow predict. Should improve
performance
---
py_wake/utils/tensorflow_surrogate_utils.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/py_wake/utils/tensorflow_surrogate_utils.py b/py_wake/utils/tensorflow_surrogate_utils.py
index 71cff543b..8efba343e 100644
--- a/py_wake/utils/tensorflow_surrogate_utils.py
+++ b/py_wake/utils/tensorflow_surrogate_utils.py
@@ -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):
--
GitLab