class

CrystalML::Regression::RandomForestRegression

Inherits CrystalML::Regressor / CrystalML::SupervisedEstimator / CrystalML::BaseEstimator / Reference / Object

Constructors

new(n_trees : Int32 = 100, max_depth : Int32 = 10, min_size : Int32 = 2)
Source

Instance methods

fit(data : Tensor(Float64, CPU(Float64)), target : Tensor(Float64, CPU(Float64)))
Source
max_depth
Source
max_depth=(max_depth : Int32)
Source
min_size
Source
min_size=(min_size : Int32)
Source
n_trees
Source
n_trees=(n_trees : Int32)
Source
predict(data : Tensor(Float64, CPU(Float64))) : Tensor(Float64, CPU(Float64))
Source
trees
Source
trees=(trees : Array(DecisionTreeRegression))
Source