Skip to content

Commit a74a33b

Browse files
committed
Document the output of predict()
1 parent ada0950 commit a74a33b

File tree

1 file changed

+8
-0
lines changed
  • zooprocess_multiple_classifier

1 file changed

+8
-0
lines changed

zooprocess_multiple_classifier/api.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,14 @@ def get_predict_args():
105105
def predict(**kwargs):
106106
"""
107107
Predict the classification of an object
108+
109+
Args:
110+
See get_predict_args() above.
111+
112+
Returns:
113+
A float, in [0,1]: the probability for the image to be a multiple. A natural
114+
threshold to classify it as multiple is 0.5 but lowering this threshold can
115+
increase the recall of multiples, at the expense of precision.
108116
"""
109117

110118
# read image

0 commit comments

Comments
 (0)