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