Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
mednet
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Package Registry
Model registry
Operate
Environments
Terraform modules
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
medai
software
mednet
Commits
1c1600a8
Commit
1c1600a8
authored
8 months ago
by
André Anjos
Browse files
Options
Downloads
Patches
Plain Diff
[libs.segmentation.engine.evaluator] Fix doc strings
parent
40aa5aba
No related branches found
No related tags found
1 merge request
!46
Create common library
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
src/mednet/libs/segmentation/engine/evaluator.py
+8
-0
8 additions, 0 deletions
src/mednet/libs/segmentation/engine/evaluator.py
with
8 additions
and
0 deletions
src/mednet/libs/segmentation/engine/evaluator.py
+
8
−
0
View file @
1c1600a8
...
@@ -532,25 +532,33 @@ def run(
...
@@ -532,25 +532,33 @@ def run(
* ``counts``: dictionary where keys are thresholds, and values are
* ``counts``: dictionary where keys are thresholds, and values are
sequence of integers containing the TP, FP, TN, FN (in this order).
sequence of integers containing the TP, FP, TN, FN (in this order).
* ``auc_score``: a float indicating the area under the ROC curve
* ``auc_score``: a float indicating the area under the ROC curve
for the split. It is calculated using a trapezoidal rule.
for the split. It is calculated using a trapezoidal rule.
* ``average_precision_score``: a float indicating the area under the
* ``average_precision_score``: a float indicating the area under the
precision-recall curve, calculated using a rectangle rule.
precision-recall curve, calculated using a rectangle rule.
* ``curves``: dictionary with 2 keys:
* ``curves``: dictionary with 2 keys:
* ``roc``: dictionary with 3 keys:
* ``roc``: dictionary with 3 keys:
* ``fpr``: a list of floats with the false-positive rate
* ``fpr``: a list of floats with the false-positive rate
* ``tpr``: a list of floats with the true-positive rate
* ``tpr``: a list of floats with the true-positive rate
* ``thresholds``: a list of thresholds uniformily separated by
* ``thresholds``: a list of thresholds uniformily separated by
``steps``, at which both ``fpr`` and ``tpr`` are evaluated.
``steps``, at which both ``fpr`` and ``tpr`` are evaluated.
* ``precision_recall``: a dictionary with 3 keys:
* ``precision_recall``: a dictionary with 3 keys:
* ``precision``: a list of floats with the precision
* ``precision``: a list of floats with the precision
* ``recall``: a list of floats with the recall
* ``recall``: a list of floats with the recall
* ``thresholds``: a list of thresholds uniformily separated by
* ``thresholds``: a list of thresholds uniformily separated by
``steps``, at which both ``precision`` and ``recall`` are
``steps``, at which both ``precision`` and ``recall`` are
evaluated.
evaluated.
* ``threshold_a_priori``: boolean indicating if the threshold for unary
* ``threshold_a_priori``: boolean indicating if the threshold for unary
metrics where computed with a threshold chosen a priori or a
metrics where computed with a threshold chosen a priori or a
posteriori in this split.
posteriori in this split.
* ``<metric-name>``: a float representing the supported metric at the
* ``<metric-name>``: a float representing the supported metric at the
threshold that maximizes ``metric``. There will be one entry of this
threshold that maximizes ``metric``. There will be one entry of this
type for each of the :py:obj:`SUPPORTED_METRIC_TYPE`
'
s.
type for each of the :py:obj:`SUPPORTED_METRIC_TYPE`
'
s.
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment