try to avoid possible memory leaks
This commit is contained in:
parent
b438cd4b3f
commit
71c6ff18c4
@ -252,9 +252,9 @@ The `CNNPredictionModel` is a non-linear regression based on `Tensorflow` which
|
||||
"batch_size": 64,
|
||||
"epochs": 10,
|
||||
"verbose": "auto",
|
||||
"shuffle": false
|
||||
"shuffle": false,
|
||||
"workers": 1,
|
||||
"use_multiprocessing": False
|
||||
"use_multiprocessing": false
|
||||
}
|
||||
```
|
||||
|
||||
|
@ -133,6 +133,7 @@ class CNNPredictionModel(BaseTensorFlowModel):
|
||||
else:
|
||||
data = dk.data_dictionary["prediction_features"]
|
||||
data = tf.expand_dims(data, axis=0)
|
||||
data = tf.convert_to_tensor(data)
|
||||
predictions = self.model(data, training=False)
|
||||
|
||||
predictions = predictions[:, 0, 0]
|
||||
|
Loading…
Reference in New Issue
Block a user