Skip to content

Commit

Permalink
Fix docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
bofenghuang authored Apr 24, 2023
1 parent 658c9ee commit 022ac73
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/setfit/modeling.py
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,7 @@ class SetFitHead(models.Dense):
out_features (`int`, defaults to `2`):
The number of targets. If set `out_features` to 1 for binary classification, it will be changed to 2 as 2-class classification.
temperature (`float`, defaults to `1.0`):
A logits' scaling factor. Higher values makes the model less confident and higher values makes
A logits' scaling factor. Higher values make the model less confident and lower values make
it more confident.
eps (`float`, defaults to `1e-5`):
A value for numerical stability when scaling logits.
Expand Down

0 comments on commit 022ac73

Please sign in to comment.