Update eis_toolkit/deep_learning/mlp_function.py

tha' s ok to me Co-authored-by: Niko Aarnio <[email protected]>
GispoCoding · Oct 26, 2023 · 498a8b4 · 498a8b4
1 parent 9fcf984
commit 498a8b4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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.