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