class CrystalML::Classification::RandomForestClassifier

• CrystalML::Classification::RandomForestClassifier
• CrystalML::SupervisedEstimator
• CrystalML::BaseEstimator
• Reference
• Object

Included Modules

• CrystalML::Classifier

Defined in:

classification/random_forest_classifier.cr

Constructors

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

Instance Method Summary

• #build_tree(data : Tensor(Float64, CPU(Float64)), target : Tensor(Float64, CPU(Float64))) : DecisionTreeClassifier
• #fit(data : Tensor(Float64, CPU(Float64)), target : Tensor(Float64, CPU(Float64)))
• #max_depth : Int32
• #max_depth=(max_depth : Int32)
• #min_size : Int32
• #min_size=(min_size : Int32)
• #n_trees : Int32
• #n_trees=(n_trees : Int32)
• #predict(data : Tensor(Float64, CPU(Float64))) : Tensor(Float64, CPU(Float64))
• #trees : Array(DecisionTreeClassifier)
• #trees=(trees : Array(DecisionTreeClassifier))

Constructor Detail

Instance Method Detail