class Accumulator

继承自: object

累积计算出的 embedding 和对应的 target，用于度量指标和评估器。

reset()

重置累积状态

用于重置累加器状态。

set_filled()

阻止进一步的累积

update(**kwargs) → None

累积批次

参数:

**kwargs (- embedding 和计算标签所需的其他对象。例如，) –

• labels (对于基于 pair 的任务，它就是) –

• pairs –

• is (subgroups，对于基于 group 的任务，它) –

• groups. –

• property embeddings

将 embedding 列表连接成 Tensor

有助于避免在累积期间对每个批次的 embedding 进行连接。相反，只在调用时连接。

返回:

torch.Tensor – embedding 批次

property filled: bool

累加器的状态

bool - 表示累加器是否仍然可以累积值，或者它是否已经

torch.Tensor – embedding 批次

填满

property state: Dict[str, Tensor]

累积的状态

Dict[str, torch.Tensor] - 包含相应字段名称和累积值的字典

torch.Tensor – embedding 批次

values

下一页 上一页