reblackened, reran notebooks
@@ -439,8 +439,7 @@ class Experiment(object):
max_size, n_classes = preds.shape
# pre-allocate memory for all predictions and targets
all_preds = np.nan * np.ones(
- (n_batches * max_size, n_classes),
- dtype=np.float32,
+ (n_batches * max_size, n_classes), dtype=np.float32
)
else:
assert len(preds.shape) == 3
