We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7682e89 commit ddc49ccCopy full SHA for ddc49cc
lib/axon/losses.ex
@@ -83,10 +83,10 @@ defmodule Axon.Losses do
83
* `:reduction` - reduction mode. One of `:mean`, `:sum`, or `:none`.
84
Defaults to `:none`.
85
86
- * `:negative_weights` - class weight for `0` class useful for scaling loss
+ * `:negative_weight` - class weight for `0` class useful for scaling loss
87
by importance of class. Defaults to `1.0`.
88
89
- * `:positive_weights` - class weight for `1` class useful for scaling loss
+ * `:positive_weight` - class weight for `1` class useful for scaling loss
90
91
92
* `:from_logits` - whether `y_pred` is a logits tensor. Defaults to `false`.
0 commit comments