-
Notifications
You must be signed in to change notification settings - Fork 126
Open
Labels
enhancementNew feature or requestNew feature or request
Description
Please add online hard example mining (OHEM) based on BCE, L1, L2 and other losses for binary classification.
Loss = SumOfPositiveClassLosses / + alpha* SumOfNegativeClassLosses / + beta * SumOfHARDNegativeClassLosses / <number of hard negative class examples>
hard negatives
are top K negative class elements with highest loss value.
alpha
, beta
, K
are external parameters.
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or request