From d268a39cbcc94b9742c03c48f8f48fec969ec400 Mon Sep 17 00:00:00 2001 From: KowshikB03 <200238038+KowshikB03@users.noreply.github.com> Date: Thu, 6 Mar 2025 16:31:15 +0000 Subject: [PATCH] Automatic `pre-commit` fixes --- aeon/classification/deep_learning/_cnn.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/aeon/classification/deep_learning/_cnn.py b/aeon/classification/deep_learning/_cnn.py index 858b780232..a3deda7b71 100644 --- a/aeon/classification/deep_learning/_cnn.py +++ b/aeon/classification/deep_learning/_cnn.py @@ -213,10 +213,11 @@ def build_model(self, input_shape, n_classes, **kwargs): """ import numpy as np import tensorflow as tf + # Change activation to softmax only if it's a multi-class problem and no activation is provided. if self.activation is None: self.activation = "softmax" if n_classes > 2 else "sigmoid" - print("activation:",self.activation,flush=True) + print("activation:", self.activation, flush=True) rng = check_random_state(self.random_state) self.random_state_ = rng.randint(0, np.iinfo(np.int32).max) tf.keras.utils.set_random_seed(self.random_state_)