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 }