Skip to content

Commit 64a6a9f

Browse files
committed
classification integration docs and example
1 parent b3c0d85 commit 64a6a9f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/classifier_integration_ex.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,9 @@
1414
transaction_features = torch.rand(1, 5)
1515

1616
# Define integration options.
17-
# Only probabilities above 0.4 are considered for adjustment.
17+
# Only probabilities above 0.5 are considered for adjustment.
1818
interface_options = pr.ModelInterfaceOptions(
19-
threshold=0.5, # Only process probabilities above 0.6
19+
threshold=0.5, # Only process probabilities above 0.5
2020
set_lower_bound=True, # For high confidence, adjust the lower bound.
2121
set_upper_bound=False, # Keep the upper bound unchanged.
2222
snap_value=1.0 # Use 1.0 as the snap value.

0 commit comments

Comments
 (0)