RemoveAccidentalHits 类keras_rs.layers.RemoveAccidentalHits(
activity_regularizer=None,
trainable=True,
dtype=None,
autocast=True,
name=None,
**kwargs
)
将意外负例的 logits 置零。
将与该行中的正例具有相同 ID 的负例候选的 logits 置零。
示例
# Create layer with the configured number of hard negatives to mine.
remove_accidental_hits = keras_rs.layers.RemoveAccidentalHits()
# This will zero the logits of negative candidates that have the same ID as
# the positive candidate from `labels` so as to not negatively impact the
# true positive.
logits = remove_accidental_hits(logits, labels, candidate_ids)
call 方法RemoveAccidentalHits.call(logits: Any, labels: Any, candidate_ids: Any)
将选定的 logits 置零。
对于批次中的每一行，将与该行中的正例具有相同 ID 的负例候选的 logits 置零。
参数
[batch_size, num_candidates]，但也可以有更多维度或为 1D [num_candidates]。logits 相同。[num_candidates] 或 [batch_size, num_candidates] 或有更多维度，只要它们与 labels 的最后一个维度匹配即可。返回
修改后的 logits，形状与输入 logits 相同。