Skip to content
Snippets Groups Projects
Commit 45b10064 authored by tomrink's avatar tomrink
Browse files

minor

parent 3efe6ca1
Branches
No related tags found
No related merge requests found
......@@ -161,6 +161,7 @@ class IcingIntensityNN:
self.test_labels = []
self.test_preds = []
self.test_probs
self.learningRateSchedule = None
self.num_data_samples = None
......@@ -717,6 +718,7 @@ class IcingIntensityNN:
labels = np.concatenate(self.test_labels)
preds = np.concatenate(self.test_preds)
self.test_probs = preds
preds = np.where(preds > 0.5, 1, 0)
self.test_labels = labels
......
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment