RUCAIBox / RecBole

A unified, comprehensive and efficient recommendation library
https://recbole.io/
MIT License
3.48k stars 616 forks source link

[💡SUG] 请问Loss函数有哪几种类型,如何修改Loss函数 #1850

Open Superboshen opened 1 year ago

Superboshen commented 1 year ago

您好,我在训练FM模型,想用focal loss函数,但在配置文档中没有找到loss的配置

Ethan-TZ commented 1 year ago

@Superboshen 您好,我们目前没有提供Focal Loss配置。对于context-aware类模型,我们目前只支持BCE Loss。 当然您可以自己实现Focal Loss,并且将模型的变量设置为 self.loss=[Your Implemented Focal Loss] 即可。