class CrystalML::Classification::DecisionTreeClassifier

Included Modules

Defined in:

classification/decision_tree_classifier.cr

Instance Method Summary

Instance methods inherited from module CrystalML::Classifier

predict(data : Tensor(Float64, CPU(Float64))) : Tensor(Float64, CPU(Float64))
predict(data : Array(Array(Float64)) | Crysda::DataFrame) : Tensor(Float64, CPU(Float64)) predict

Instance methods inherited from class CrystalML::Tree::DecisionTree

build_tree(data : Tensor(Float64, CPU(Float64)), target : Tensor(Float64, CPU(Float64)), depth : Int32) : Node build_tree, fit(data : Tensor(Float64, CPU(Float64)), target : Tensor(Float64, CPU(Float64))) fit, max_depth : Int32 max_depth, max_depth=(max_depth : Int32) max_depth=, min_size : Int32 min_size, min_size=(min_size : Int32) min_size=, predict(data : Tensor(Float64, CPU(Float64))) : Tensor(Float64, CPU(Float64)) predict, print_tree print_tree, print_tree_recursive(node : Tree::Node | Nil, depth : Int32) print_tree_recursive, root : Node | Nil root, root=(root : Node | Nil) root=

Constructor methods inherited from class CrystalML::Tree::DecisionTree

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

Instance methods inherited from class CrystalML::SupervisedEstimator

fit(data : Tensor(Float64, CPU(Float64)), target : Tensor(Float64, CPU(Float64)))
fit(data : Tensor(Float64, CPU(Float64)) | Array(Array(Float64)) | Crysda::DataFrame, target : Tensor(Float64, CPU(Float64)) | Array(Array(Float64)) | Array(Float64) | Crysda::DataFrame) fit

Instance Method Detail

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

[View source]