Versions in this module Expand all Collapse all v1 v1.0.0 Nov 4, 2025 Changes in this version + type DecisionTree struct + Criterion string + MaxDepth int + MinSamples int + func NewDecisionTreeClassifier(maxDepth, minSamples int, criterion string) *DecisionTree + func NewDecisionTreeRegressor(maxDepth, minSamples int) *DecisionTree + func (dt *DecisionTree) FeatureImportances() []float64 + func (dt *DecisionTree) Fit(X *dataframe.DataFrame, y *seriesPkg.Series[any]) error + func (dt *DecisionTree) Predict(X *dataframe.DataFrame) (*seriesPkg.Series[any], error)
Go to main page