Skip to content

Commit

Permalink
Automatic pre-commit fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
KowshikB03 authored and github-actions[bot] committed Mar 6, 2025
1 parent ca7e200 commit d268a39
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion aeon/classification/deep_learning/_cnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -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_)
Expand Down

0 comments on commit d268a39

Please sign in to comment.