class

CrystalML::Regression::DecisionTreeRegression

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

Instance methods

build_tree(data : Tensor(Float64, CPU(Float64)), target : Tensor(Float64, CPU(Float64)), depth : Int32) : Tree::Node
Source