Skip to content

Commit

Permalink
DOC Fix time complexity of MLP (scikit-learn#28592)
Browse files Browse the repository at this point in the history
Co-authored-by: Johann <[email protected]>
  • Loading branch information
jpienaar-tuks and Johann committed May 18, 2024
1 parent 48669a5 commit d88b413
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion doc/modules/neural_networks_supervised.rst
Original file line number Diff line number Diff line change
Expand Up @@ -229,7 +229,7 @@ Complexity
Suppose there are :math:`n` training samples, :math:`m` features, :math:`k`
hidden layers, each containing :math:`h` neurons - for simplicity, and :math:`o`
output neurons. The time complexity of backpropagation is
:math:`O(n\cdot m \cdot h^k \cdot o \cdot i)`, where :math:`i` is the number
:math:`O(i \cdot n \cdot (m \cdot h + (k - 1) \cdot h \cdot h + h \cdot o))`, where :math:`i` is the number
of iterations. Since backpropagation has a high time complexity, it is advisable
to start with smaller number of hidden neurons and few hidden layers for
training.
Expand Down

0 comments on commit d88b413

Please sign in to comment.