class

CrystalML::Classification::DecisionTreeClassifier

Inherits CrystalML::Classifier / 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