1. hinge loss
参考链接:https://zhuanlan.zhihu.com/p/20945670?refer=intelligentunit
2. pytorch 实现
https://github.com/kazuto1011/svm-pytorch
3. 代码
loss计算:
loss = torch.mean(torch.clamp(1-fea.t()*label, min=0))
loss += c*torch.mean(model.fea.weight**2)
svm分类器:
self.fea = nn.Linear(2, 1)