abstract class CrystalML::Tree::DecisionTree

Direct Known Subclasses

Defined in:

utils/tree.cr

Constructors

Instance Method Summary

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

Constructor Detail

def self.new(max_depth : Int32 = 10, min_size : Int32 = 2) #

[View source]

Instance Method Detail

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

[View source]
def fit(data : Tensor(Float64, CPU(Float64)), target : Tensor(Float64, CPU(Float64))) #

[View source]
def max_depth : Int32 #

[View source]
def max_depth=(max_depth : Int32) #

[View source]
def min_size : Int32 #

[View source]
def min_size=(min_size : Int32) #

[View source]
def predict(data : Tensor(Float64, CPU(Float64))) : Tensor(Float64, CPU(Float64)) #

[View source]
def print_tree #

[View source]
def print_tree_recursive(node : Tree::Node | Nil, depth : Int32) #

[View source]
def root : Node | Nil #

[View source]
def root=(root : Node | Nil) #

[View source]