Skip to content
Snippets Groups Projects
Commit 483c3954 authored by Noordijk, Ben's avatar Noordijk, Ben
Browse files

Built-in predict function of CNN now directly calls model to be faster

parent 280a7b80
No related branches found
No related tags found
No related merge requests found
......@@ -155,7 +155,7 @@ class NeuralNetwork(object):
truncating='post',
dtype='float32'), -1)
posteriors = self.model.predict(x_pad)
posteriors = self.model(x_pad)
y_hat = posteriors > self.threshold
y_out = np.zeros(len(x), dtype=int)
for i, yh in enumerate(y_hat):
......
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