lib/rumale/ensemble/extra_trees_regressor.rb in rumale-0.12.0 vs lib/rumale/ensemble/extra_trees_regressor.rb in rumale-0.12.1
- old
+ new
@@ -127,10 +127,10 @@
def plant_tree(rnd_seed)
Tree::ExtraTreeRegressor.new(
criterion: @params[:criterion], max_depth: @params[:max_depth],
max_leaf_nodes: @params[:max_leaf_nodes], min_samples_leaf: @params[:min_samples_leaf],
- max_features: @params[:max_features], random_seed:rnd_seed
+ max_features: @params[:max_features], random_seed: rnd_seed
)
end
end
end
end