From 498a8b4e06338ec67e53d95d5500e82daad6c3cd Mon Sep 17 00:00:00 2001 From: Luca Zelioli Date: Thu, 26 Oct 2023 12:29:41 +0300 Subject: [PATCH] Update eis_toolkit/deep_learning/mlp_function.py tha' s ok to me Co-authored-by: Niko Aarnio <113038549+nmaarnio@users.noreply.github.com> --- eis_toolkit/deep_learning/mlp_function.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/eis_toolkit/deep_learning/mlp_function.py b/eis_toolkit/deep_learning/mlp_function.py index b7380ad1..7347787c 100644 --- a/eis_toolkit/deep_learning/mlp_function.py +++ b/eis_toolkit/deep_learning/mlp_function.py @@ -33,8 +33,10 @@ def train_evaluate_predict_with_mlp( alpha: floating point represent regularization. hidden_layer_sizes: It represents the number of neurons in the ith hidden layer. random_state: random state for repeatability of results. + Return: - a numpy array with prediction (class if is_class_probability is set to false otherwise it return probability). + A Numpy array with prediction (class if is_class_probability is set to false otherwise it return probability). + Raises: InvalidDatasetException: When the dataset is None. InvalidArgumentTypeException when the function try to make probability and the threshold is None.