Skip to content

return of nnPU loss function may be mistaken #3

Description

@takehiro-yamane

In 'loss.py' , line 37-38

        if negative_risk < -self.beta and self.nnPU:
            return -self.gamma * negative_risk

Is this correct?

I think the following is correct.

        if negative_risk < -self.beta and self.nnPU:
            return positive_risk

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions