lib/rumale/ensemble/random_forest_regressor.rb in rumale-ensemble-0.25.0 vs lib/rumale/ensemble/random_forest_regressor.rb in rumale-ensemble-0.26.0
- old
+ new
@@ -79,10 +79,10 @@
::Rumale::Validation.check_sample_size(x, y)
# Initialize some variables.
n_samples, n_features = x.shape
@params[:max_features] = Math.sqrt(n_features).to_i if @params[:max_features].nil?
- @params[:max_features] = [[1, @params[:max_features]].max, n_features].min
+ @params[:max_features] = [[1, @params[:max_features]].max, n_features].min # rubocop:disable Style/ComparableClamp
single_target = y.shape[1].nil?
sub_rng = @rng.dup
rngs = Array.new(@params[:n_estimators]) { Random.new(sub_rng.rand(::Rumale::Ensemble::Value::SEED_BASE)) }
# Construct forest.
@estimators =