From ceea333abf30e008ae79409ebbc55005918195db Mon Sep 17 00:00:00 2001 From: Aadya Chinubhai <77720426+aadya940@users.noreply.github.com> Date: Mon, 11 Nov 2024 12:32:37 +0530 Subject: [PATCH] Update _mlp.py --- aeon/networks/_mlp.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/aeon/networks/_mlp.py b/aeon/networks/_mlp.py index 50c97a175b..8e4ef3b41d 100644 --- a/aeon/networks/_mlp.py +++ b/aeon/networks/_mlp.py @@ -3,7 +3,7 @@ __maintainer__ = ["hadifawaz1999"] -from typing import List, Union +from typing import Union from aeon.networks.base import BaseDeepLearningNetwork @@ -17,11 +17,11 @@ class MLPNetwork(BaseDeepLearningNetwork): ---------- n_layers : int, optional (default=3) The number of dense layers in the MLP. - n_units : Union[int, List[int]], optional (default=200) + n_units : Union[int, list[int]], optional (default=200) Number of units in each dense layer. - activation : Union[str, List[str]], optional (default='relu') + activation : Union[str, list[str]], optional (default='relu') Activation function(s) for each dense layer. - dropout_rate : Union[int, float, List[Union[int, float]]], optional (default=None) + dropout_rate : Union[int, float, list[Union[int, float]]], optional (default=None) Dropout rate(s) for each dense layer. If None, a default rate of 0.2 is used. Notes