pytorch
文章平均质量分 90
zzllg
这个作者很懒,什么都没留下…
展开
-
【机器学习】focalloss原理以及pytorch实现
最近在做一个分类项目,发现很多“难样本”比较不好处理又特别重要,想试试FocalLoss。没找到pytorch相关实现,本来想研究pytorch的cross_entropy源码,稍微改一下(怕手残自己写的loss效率比较低),但是发现有点复杂,我的任务比较简单,改那玩意有点累。我们知道,对于二分类:cross_entropy(y,y^)=−ylogy^−(1−y)log1−y^cross\_...原创 2020-04-19 21:55:17 · 1464 阅读 · 8 评论 -
【机器学习】关于pytorch中的分类损失函数cross_entropy使用log_softmax
最近在做一个分类项目,发现很多“难样本”比较不好处理,想试试FocalLoss。没找到pytorch相关实现,就研究起cross_entropy源码,想手动改一下。def cross_entropy(input, target, weight=None, size_average=None, ignore_index=-100, reduce=None, re...原创 2020-04-18 17:11:47 · 630 阅读 · 0 评论