lib/rumale/kernel_machine/kernel_svc.rb in rumale-0.18.1 vs lib/rumale/kernel_machine/kernel_svc.rb in rumale-0.18.2
- old
+ new
@@ -159,30 +159,9 @@
probs[true, 1] = 1.0 / (Numo::NMath.exp(@prob_param[0] * decision_function(x) + @prob_param[1]) + 1.0)
probs[true, 0] = 1.0 - probs[true, 1]
probs
end
- # Dump marshal data.
- # @return [Hash] The marshal data about KernelSVC.
- def marshal_dump
- { params: @params,
- weight_vec: @weight_vec,
- prob_param: @prob_param,
- classes: @classes,
- rng: @rng }
- end
-
- # Load marshal data.
- # @return [nil]
- def marshal_load(obj)
- @params = obj[:params]
- @weight_vec = obj[:weight_vec]
- @prob_param = obj[:prob_param]
- @classes = obj[:classes]
- @rng = obj[:rng]
- nil
- end
-
private
def partial_fit(x, bin_y)
# Initialize some variables.
n_training_samples = x.shape[0]