lib/svmkit/linear_model/svr.rb in svmkit-0.3.2 vs lib/svmkit/linear_model/svr.rb in svmkit-0.3.3
- old
+ new
@@ -19,15 +19,15 @@
# 1. S. Shalev-Shwartz and Y. Singer, "Pegasos: Primal Estimated sub-GrAdient SOlver for SVM," Proc. ICML'07, pp. 807--814, 2007.
class SVR
include Base::BaseEstimator
include Base::Regressor
- # Return the weight vector for SVC.
+ # Return the weight vector for SVR.
# @return [Numo::DFloat] (shape: [n_outputs, n_features])
attr_reader :weight_vec
- # Return the bias term (a.k.a. intercept) for SVC.
+ # Return the bias term (a.k.a. intercept) for SVR.
# @return [Numo::DFloat] (shape: [n_outputs])
attr_reader :bias_term
# Return the random generator for performing random sampling.
# @return [Random]
@@ -102,10 +102,10 @@
SVMKit::Validation.check_sample_array(x)
x.dot(@weight_vec.transpose) + @bias_term
end
# Dump marshal data.
- # @return [Hash] The marshal data about SVC.
+ # @return [Hash] The marshal data about SVR.
def marshal_dump
{ params: @params,
weight_vec: @weight_vec,
bias_term: @bias_term,
rng: @rng }