Index
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
Trainerthat 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
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
Mapfor 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.
All Classes and Interfaces|All Packages|Serialized Form