class

CrystalML::Tree::DecisionTree

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

Constructors

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

Instance methods

build_tree(data : Tensor(Float64, CPU(Float64)), target : Tensor(Float64, CPU(Float64)), depth : Int32) : Node
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
predict(data : Tensor(Float64, CPU(Float64))) : Tensor(Float64, CPU(Float64))
Source
print_tree_recursive(node : Tree::Node | Nil, depth : Int32)
Source
root
Source
root=(root : Node | Nil)
Source