Skip to content

Commit fc3ced1

Browse files
authored
Update technical_doc.rst
1 parent 656ca5c commit fc3ced1

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

docs/technical_doc.rst

+2-3
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,12 @@ Task inference
77
--------------
88
The funciton :func:`~mcfly.find_architecture.find_best_architecture` automatically infers the task (classification vs. regression) from the input data.
99
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:
1111

1212
* The target array contains only two unique values
1313
* 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)
1514

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
1716
depending on the inferred task (see :ref:`Other choices`).
1817

1918
Hyperparameter search

0 commit comments

Comments
 (0)