Fix docstring (#366)

* Fix docstring * Fix docstring * Fix docstring
huggingface · Apr 24, 2023 · 92e966b · 92e966b
1 parent 658c9ee
commit 92e966b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/setfit/modeling.py b/src/setfit/modeling.py
@@ -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.
@@ -176,8 +176,8 @@ def forward(
                 make sure to store embeddings under the key: 'sentence_embedding'
                 and the outputs will be under the key: 'prediction'.
             temperature (`float`, *optional*):
-                A logits' scaling factor. Higher values makes the model less
-                confident and higher values makes it more confident.
+                A logits' scaling factor. Higher values make the model less
+                confident and lower values make it more confident.
                 Will override the temperature given during initialization.
         Returns:
             [`Dict[str, torch.Tensor]` or `Tuple[torch.Tensor]`]