search

shenweichen / DeepCTR-Torch

【PyTorch】Easy-to-use,Modular and Extendible package of deep-learning based CTR models.
https://deepctr-torch.readthedocs.io/en/latest/index.html
Apache License 2.0
2.95k stars 696 forks source link

Custom Loss Function within xDeepFM API #286

Open andyozj opened 1 year ago

andyozj commented 1 year ago

Describe the question(问题描述) I noticed that the xDeepFM model API within this repo only supports 2 task

Does the package supports the integration of other loss function? e.g. Ranking related losses such as Hinge loss