lib/svmkit/ensemble/random_forest_classifier.rb in svmkit-0.3.2 vs lib/svmkit/ensemble/random_forest_classifier.rb in svmkit-0.3.3
- old
+ new
@@ -153,10 +153,10 @@
SVMKit::Validation.check_sample_array(x)
Numo::Int32[*Array.new(@params[:n_estimators]) { |n| @estimators[n].apply(x) }].transpose
end
# Dump marshal data.
- # @return [Hash] The marshal data about RandomForestClassifier
+ # @return [Hash] The marshal data about RandomForestClassifier.
def marshal_dump
{ params: @params, estimators: @estimators, classes: @classes,
feature_importances: @feature_importances, rng: @rng }
end