Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion docs/conf.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,11 +45,11 @@ def get_version():

extensions = [
"sphinx.ext.autodoc",
"sphinx.ext.autosummary",
"sphinx.ext.coverage",
"sphinx.ext.napoleon",
"sphinx.ext.viewcode",
"sphinx.ext.mathjax",
"autodocsumm",
]

# Add any paths that contain templates here, relative to this directory.
Expand Down
27 changes: 25 additions & 2 deletions docs/metrics.rst
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,29 @@

Functional metrics
~~~~~~~~~~~~~~~~~~

.. currentmodule:: segmentation_models_pytorch.metrics.functional

.. autosummary::

get_stats
fbeta_score
f1_score
iou_score
accuracy
precision
recall
sensitivity
specificity
balanced_accuracy
positive_predictive_value
negative_predictive_value
false_negative_rate
false_positive_rate
false_discovery_rate
false_omission_rate
positive_likelihood_ratio
negative_likelihood_ratio

.. automodule:: segmentation_models_pytorch.metrics.functional
:members:
:autosummary:
:members:
1 change: 0 additions & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,6 @@ dynamic = ['version']

[project.optional-dependencies]
docs = [
'autodocsumm',
'huggingface-hub',
'six',
'sphinx',
Expand Down
3 changes: 1 addition & 2 deletions requirements/docs.txt
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
autodocsumm==0.2.14
huggingface-hub==1.7.1
six==1.17.0
sphinx==8.2.3
sphinx==9.1.0
sphinx-book-theme==1.2.0
Loading