Skip to content

Commit

Permalink
Fix typos in docs about options (#566)
Browse files Browse the repository at this point in the history
  • Loading branch information
preciz authored Apr 9, 2024
1 parent 7682e89 commit ddc49cc
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions lib/axon/losses.ex
Original file line number Diff line number Diff line change
Expand Up @@ -83,10 +83,10 @@ defmodule Axon.Losses do
* `:reduction` - reduction mode. One of `:mean`, `:sum`, or `:none`.
Defaults to `:none`.
* `:negative_weights` - class weight for `0` class useful for scaling loss
* `:negative_weight` - class weight for `0` class useful for scaling loss
by importance of class. Defaults to `1.0`.
* `:positive_weights` - class weight for `1` class useful for scaling loss
* `:positive_weight` - class weight for `1` class useful for scaling loss
by importance of class. Defaults to `1.0`.
* `:from_logits` - whether `y_pred` is a logits tensor. Defaults to `false`.
Expand Down

0 comments on commit ddc49cc

Please sign in to comment.