class

CrystalML::Classification::RandomForestClassifier

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

Constructors

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

Instance methods

build_tree(data : Tensor(Float64, CPU(Float64)), target : Tensor(Float64, CPU(Float64))) : DecisionTreeClassifier
Source
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(DecisionTreeClassifier))
Source