You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: docs/technical_doc.rst
+2-3
Original file line number
Diff line number
Diff line change
@@ -7,13 +7,12 @@ Task inference
7
7
--------------
8
8
The funciton :func:`~mcfly.find_architecture.find_best_architecture` automatically infers the task (classification vs. regression) from the input data.
9
9
The task is inferred from *y_train* and *y_val* if both are not `None`. If *X_train* and *X_val* are Tensorflow datasets or generators,
10
-
the task is inferred from the targets of the first batch. The function performs classification if the target complies with one-hot encoding, specifically if all three conditions are met:
10
+
the task is inferred from the targets of the first batch. The function performs classification if the target complies with one-hot encoding, specifically if both conditions are met:
11
11
12
12
* The target array contains only two unique values
13
13
* The unique values are 0 and 1
14
-
* The second dimension of the target array has length > 1 (i.e., there are at least two classes)
15
14
16
-
If any condition is not met, regression is performed. If :func:`~mcfly.find_architecture.find_best_architecture` received `metric=None`, a default metric is chosen
15
+
If either condition is not met, regression is performed. If :func:`~mcfly.find_architecture.find_best_architecture` received `metric=None`, a default metric is chosen
17
16
depending on the inferred task (see :ref:`Other choices`).
0 commit comments