Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

loss函数 #31

Open
topDreamer opened this issue Sep 4, 2021 · 1 comment
Open

loss函数 #31

topDreamer opened this issue Sep 4, 2021 · 1 comment

Comments

@topDreamer
Copy link

非常好的工作!
有个疑问,请问假设最后构建出来矩阵为:[batch_size, seq_lenth, seq_lenth, num_labels],是否考虑过使用多标签分类的loss来优化呢?

@juntaoy
Copy link
Owner

juntaoy commented Sep 6, 2021

常用的数据集暂时没有这样的需求所以我也没有试过,估计是可以的你把loss改成sigmoid应该就可以的。我之前的mention detecter的论文里面用的就是sigmoid函数(不过是binary的,num_labels = 1) 所以应该可行的。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants