Index

B C E G I L M O T V 
All Classes and Interfaces|All Packages|Serialized Form

B

buildRandomTree(int[], SplittableRandom) - Method in class org.tribuo.classification.dtree.impl.ClassifierTrainingNode
Builds a CART tree with randomly chosen split points.
buildTree(int[], SplittableRandom, boolean) - Method in class org.tribuo.classification.dtree.impl.ClassifierTrainingNode
Builds a tree according to CART (as it does not do multi-way splits on categorical values like C4.5).

C

CART - Enum constant in enum org.tribuo.classification.dtree.CARTClassificationOptions.TreeType
Builds a CART model.
CARTClassificationOptions - Class in org.tribuo.classification.dtree
Options for building a classification tree trainer.
CARTClassificationOptions() - Constructor for class org.tribuo.classification.dtree.CARTClassificationOptions
 
CARTClassificationOptions.ImpurityType - Enum in org.tribuo.classification.dtree
The impurity algorithm.
CARTClassificationOptions.TreeType - Enum in org.tribuo.classification.dtree
Type of decision tree algorithm.
CARTClassificationTrainer - Class in org.tribuo.classification.dtree
A Trainer that uses an approximation of the CART algorithm to build a decision tree.
CARTClassificationTrainer() - Constructor for class org.tribuo.classification.dtree.CARTClassificationTrainer
Creates a CART Trainer.
CARTClassificationTrainer(int) - Constructor for class org.tribuo.classification.dtree.CARTClassificationTrainer
Creates a CART trainer.
CARTClassificationTrainer(int, float, boolean, long) - Constructor for class org.tribuo.classification.dtree.CARTClassificationTrainer
Creates a CART Trainer.
CARTClassificationTrainer(int, float, float, float, boolean, LabelImpurity, long) - Constructor for class org.tribuo.classification.dtree.CARTClassificationTrainer
Creates a CART Trainer.
CARTClassificationTrainer(int, float, float, float, LabelImpurity, long) - Constructor for class org.tribuo.classification.dtree.CARTClassificationTrainer
Creates a CART Trainer.
CARTClassificationTrainer(int, float, long) - Constructor for class org.tribuo.classification.dtree.CARTClassificationTrainer
Creates a CART Trainer.
cartImpurity - Variable in class org.tribuo.classification.dtree.CARTClassificationOptions
Impurity measure to use.
cartMaxDepth - Variable in class org.tribuo.classification.dtree.CARTClassificationOptions
Maximum depth in the decision tree.
cartMinChildWeight - Variable in class org.tribuo.classification.dtree.CARTClassificationOptions
Minimum child weight.
cartMinImpurityDecrease - Variable in class org.tribuo.classification.dtree.CARTClassificationOptions
Minimum impurity decrease.
cartOptions - Variable in class org.tribuo.classification.dtree.TrainTest.TrainTestOptions
The CART trainer options.
cartPrintTree - Variable in class org.tribuo.classification.dtree.CARTClassificationOptions
Prints the decision tree.
cartRandomSplit - Variable in class org.tribuo.classification.dtree.CARTClassificationOptions
Whether to choose split points for features at random.
cartSeed - Variable in class org.tribuo.classification.dtree.CARTClassificationOptions
RNG seed.
cartSplitFraction - Variable in class org.tribuo.classification.dtree.CARTClassificationOptions
Fraction of features in split.
cartTreeAlgorithm - Variable in class org.tribuo.classification.dtree.CARTClassificationOptions
Tree algorithm to use (options are CART).
ClassifierTrainingNode - Class in org.tribuo.classification.dtree.impl
A decision tree node used at training time.
ClassifierTrainingNode(LabelImpurity, Dataset<Label>, AbstractTrainingNode.LeafDeterminer) - Constructor for class org.tribuo.classification.dtree.impl.ClassifierTrainingNode
Constructor which creates the inverted file.
convertTree() - Method in class org.tribuo.classification.dtree.impl.ClassifierTrainingNode
Generates a test time tree (made of SplitNode and LeafNode) from the tree rooted at this node.

E

Entropy - Class in org.tribuo.classification.dtree.impurity
A log_e entropy impurity measure.
Entropy() - Constructor for class org.tribuo.classification.dtree.impurity.Entropy
 
ENTROPY - Enum constant in enum org.tribuo.classification.dtree.CARTClassificationOptions.ImpurityType
Uses Entropy.

G

generalOptions - Variable in class org.tribuo.classification.dtree.TrainTest.TrainTestOptions
The data loading options.
getImpurity() - Method in class org.tribuo.classification.dtree.impl.ClassifierTrainingNode
 
getOptionsDescription() - Method in class org.tribuo.classification.dtree.CARTClassificationOptions
 
getOptionsDescription() - Method in class org.tribuo.classification.dtree.TrainTest.TrainTestOptions
 
getProvenance() - Method in class org.tribuo.classification.dtree.CARTClassificationTrainer
 
getProvenance() - Method in class org.tribuo.classification.dtree.impurity.Entropy
 
getProvenance() - Method in class org.tribuo.classification.dtree.impurity.GiniIndex
 
getTrainer() - Method in class org.tribuo.classification.dtree.CARTClassificationOptions
 
getWeightSum() - Method in class org.tribuo.classification.dtree.impl.ClassifierTrainingNode
 
GINI - Enum constant in enum org.tribuo.classification.dtree.CARTClassificationOptions.ImpurityType
Uses GiniIndex.
GiniIndex - Class in org.tribuo.classification.dtree.impurity
The Gini index impurity measure.
GiniIndex() - Constructor for class org.tribuo.classification.dtree.impurity.GiniIndex
 

I

impurity(double[]) - Method in interface org.tribuo.classification.dtree.impurity.LabelImpurity
Calculates the impurity assuming the inputs are counts.
impurity(float[]) - Method in interface org.tribuo.classification.dtree.impurity.LabelImpurity
Calculates the impurity assuming the input are fractional counts.
impurity(int[]) - Method in interface org.tribuo.classification.dtree.impurity.LabelImpurity
Calculates the impurity assuming the input are counts.
impurity(Map<String, Double>) - Method in interface org.tribuo.classification.dtree.impurity.LabelImpurity
Takes a Map for weighted counts.
impurityNormed(double[]) - Method in class org.tribuo.classification.dtree.impurity.Entropy
 
impurityNormed(double[]) - Method in class org.tribuo.classification.dtree.impurity.GiniIndex
 
impurityNormed(double[]) - Method in interface org.tribuo.classification.dtree.impurity.LabelImpurity
Calculates the impurity, assuming it's input is a normalized probability distribution.
impurityWeighted(double[]) - Method in interface org.tribuo.classification.dtree.impurity.LabelImpurity
Calculates the impurity assuming the inputs are weighted counts normalizing by their sum.
impurityWeighted(float[]) - Method in interface org.tribuo.classification.dtree.impurity.LabelImpurity
Calculates the impurity by assuming the input are weighted counts and converting them into a probability distribution by dividing by their sum.

L

LabelImpurity - Interface in org.tribuo.classification.dtree.impurity
Calculates a tree impurity score based on label counts, weighted label counts or a probability distribution.

M

main(String[]) - Static method in class org.tribuo.classification.dtree.TrainTest
Runs a TrainTest CLI.
mkTrainingNode(Dataset<Label>, AbstractTrainingNode.LeafDeterminer) - Method in class org.tribuo.classification.dtree.CARTClassificationTrainer
 

O

org.tribuo.classification.dtree - package org.tribuo.classification.dtree
Provides implementations of decision trees for classification problems.
org.tribuo.classification.dtree.impl - package org.tribuo.classification.dtree.impl
Provides internal implementation classes for classification decision trees.
org.tribuo.classification.dtree.impurity - package org.tribuo.classification.dtree.impurity
Provides classification impurity metrics for decision trees.

T

toString() - Method in class org.tribuo.classification.dtree.CARTClassificationTrainer
 
toString() - Method in class org.tribuo.classification.dtree.impurity.Entropy
 
toString() - Method in class org.tribuo.classification.dtree.impurity.GiniIndex
 
TrainTest - Class in org.tribuo.classification.dtree
Build and run a decision tree classifier for a standard dataset.
TrainTest() - Constructor for class org.tribuo.classification.dtree.TrainTest
 
TrainTest.TrainTestOptions - Class in org.tribuo.classification.dtree
Command line options.
TrainTestOptions() - Constructor for class org.tribuo.classification.dtree.TrainTest.TrainTestOptions
 

V

valueOf(String) - Static method in enum org.tribuo.classification.dtree.CARTClassificationOptions.ImpurityType
Returns the enum constant of this type with the specified name.
valueOf(String) - Static method in enum org.tribuo.classification.dtree.CARTClassificationOptions.TreeType
Returns the enum constant of this type with the specified name.
values() - Static method in enum org.tribuo.classification.dtree.CARTClassificationOptions.ImpurityType
Returns an array containing the constants of this enum type, in the order they are declared.
values() - Static method in enum org.tribuo.classification.dtree.CARTClassificationOptions.TreeType
Returns an array containing the constants of this enum type, in the order they are declared.
B C E G I L M O T V 
All Classes and Interfaces|All Packages|Serialized Form