lib/rumale/linear_model/svc.rb in rumale-linear_model-0.28.0 vs lib/rumale/linear_model/svc.rb in rumale-linear_model-0.28.1
- old
+ new
@@ -165,10 +165,11 @@
indices = t.gt(0)
grad = reg_param * w
if indices.count.positive?
sx = x[indices, true]
sy = y[indices]
- grad += 2.fdiv(n_samples) * sx.transpose.dot((sx.dot(w) - sy))
+ sz = z[indices]
+ grad += 2.fdiv(n_samples) * (sz - sy).dot(sx)
end
[loss, grad]
end
n_features = base_x.shape[1]