marius-team / marius

Large scale graph learning on a single machine.
https://marius-project.org
Apache License 2.0
160 stars 45 forks source link

Scope out additional loss functions #40

Closed AnzeXie closed 3 years ago

AnzeXie commented 3 years ago

Is your feature request related to a problem? Please describe. Currently, the supported loss functions we have for Marius are SoftMax and RankingLoss.

Describe the solution you'd like We can expand the set of loss functions by implementing additional loss functions to Marius. We can implement losses into 2 new source files: loss.cpp and loss.h. We can also add a new section in the configuration for loss options.

We can use the loss functions implemented by PyKeen as a reference: List: https://github.com/pykeen/pykeen#losses-7 Implementation: https://github.com/pykeen/pykeen/blob/master/src/pykeen/losses.py Documentation: https://pykeen.readthedocs.io/en/stable/reference/losses.html

JasonMoho commented 3 years ago

We should add a new doc page and describe each loss function in it and also update the configuration documentation.