public class LMTNode extends LogisticBase
Modifier and Type | Field and Description |
---|---|
double |
m_alpha
Alpha-value (for pruning) at the node
|
double |
m_numIncorrectModel
Weighted number of training examples currently misclassified by the logistic model at the node
|
double |
m_numIncorrectTree
Weighted number of training examples currently misclassified by the subtree rooted at the node
|
Constructor and Description |
---|
LMTNode(ModelSelection modelSelection,
int numBoostingIterations,
boolean fastRegression,
boolean errorOnProbabilities,
int minNumInstances,
double weightTrimBeta,
boolean useAIC)
Constructor for logistic model tree node.
|
Modifier and Type | Method and Description |
---|---|
int |
assignIDs(int lastID)
Assigns unique IDs to all nodes in the tree
|
int |
assignLeafModelNumbers(int leafCounter)
Assigns numbers to the logistic regression models at the leaves of the tree
|
void |
buildClassifier(Instances data)
Method for building a logistic model tree (only called for the root node).
|
void |
buildTree(Instances data,
SimpleLinearRegression[][] higherRegressions,
double totalInstanceWeight,
double higherNumParameters)
Method for building the tree structure.
|
void |
calculateAlphas()
Updates the alpha field for all nodes.
|
void |
cleanup()
Cleanup in order to save memory.
|
double[] |
distributionForInstance(Instance instance)
Returns the class probabilities for an instance given by the logistic model tree.
|
java.lang.String |
getModelParameters()
Returns a string describing the number of LogitBoost iterations performed at this node, the total number
of LogitBoost iterations performed (including iterations at higher levels in the tree), and the number
of training examples at this node.
|
java.util.Vector |
getNodes()
Return a list of all inner nodes in the tree
|
void |
getNodes(java.util.Vector nodeList)
Fills a list with all inner nodes in the tree
|
int |
getNumInnerNodes()
Method to count the number of inner nodes in the tree
|
int |
getNumLeaves()
Returns the number of leaves in the tree.
|
java.lang.String |
getRevision()
Returns the revision string.
|
java.lang.String |
graph()
Returns graph describing the tree.
|
boolean |
hasModels()
Returns true if the logistic regression model at this node has changed compared to the
one at the parent node.
|
double[] |
modelDistributionForInstance(Instance instance)
Returns the class probabilities for an instance according to the logistic model at the node.
|
void |
modelErrors()
Updates the numIncorrectModel field for all nodes.
|
java.lang.String |
modelsToString()
Returns a string describing the logistic regression function at the node.
|
int |
numLeaves()
Returns the number of leaves (normal count).
|
int |
numNodes()
Returns the number of nodes.
|
void |
prune(double alpha)
Prunes a logistic model tree using the CART pruning scheme, given a
cost-complexity parameter alpha.
|
int |
prune(double[] alphas,
double[] errors,
Instances test)
Method for performing one fold in the cross-validation of the cost-complexity parameter.
|
java.lang.String |
toString()
Returns a description of the logistic model tree (tree structure and logistic models)
|
void |
treeErrors()
Updates the numIncorrectTree field for all nodes.
|
getMaxIterations, getNumRegressions, getUseAIC, getUsedAttributes, getWeightTrimBeta, percentAttributesUsed, setHeuristicStop, setMaxIterations, setUseAIC, setWeightTrimBeta
classifyInstance, debugTipText, forName, getCapabilities, getDebug, getOptions, listOptions, makeCopies, makeCopy, setDebug, setOptions
public double m_alpha
public double m_numIncorrectModel
public double m_numIncorrectTree
public LMTNode(ModelSelection modelSelection, int numBoostingIterations, boolean fastRegression, boolean errorOnProbabilities, int minNumInstances, double weightTrimBeta, boolean useAIC)
modelSelection
- selection method for local splitting modelnumBoostingIterations
- sets the numBoostingIterations parameterfastRegression
- sets the fastRegression parametererrorOnProbabilities
- Use error on probabilities for stopping criterion of LogitBoost?minNumInstances
- minimum number of instances at which a node is considered for splittingpublic void buildClassifier(Instances data) throws java.lang.Exception
buildClassifier
in class LogisticBase
data
- the data to train withjava.lang.Exception
- if something goes wrongpublic void buildTree(Instances data, SimpleLinearRegression[][] higherRegressions, double totalInstanceWeight, double higherNumParameters) throws java.lang.Exception
data
- the training data passed on to this nodehigherRegressions
- An array of regression functions produced by LogitBoost at higher
levels in the tree. They represent a logistic regression model that is refined locally
at this node.totalInstanceWeight
- the total number of training exampleshigherNumParameters
- effective number of parameters in the logistic regression model built
in parent nodesjava.lang.Exception
- if something goes wrongpublic void prune(double alpha) throws java.lang.Exception
alpha
- the cost-complexity measurejava.lang.Exception
- if something goes wrongpublic int prune(double[] alphas, double[] errors, Instances test) throws java.lang.Exception
alphas
- array to hold the generated alpha-valueserrors
- array to hold the corresponding error estimatestest
- test set of that fold (to obtain error estimates)java.lang.Exception
- if something goes wrongpublic int getNumInnerNodes()
public int getNumLeaves()
public void modelErrors() throws java.lang.Exception
java.lang.Exception
public void treeErrors()
public void calculateAlphas() throws java.lang.Exception
java.lang.Exception
public java.util.Vector getNodes()
public void getNodes(java.util.Vector nodeList)
nodeList
- the list to be filledpublic boolean hasModels()
public double[] modelDistributionForInstance(Instance instance) throws java.lang.Exception
instance
- the instancejava.lang.Exception
public double[] distributionForInstance(Instance instance) throws java.lang.Exception
distributionForInstance
in class LogisticBase
instance
- the instancejava.lang.Exception
- if distribution can't be computed successfullypublic int numLeaves()
public int numNodes()
public java.lang.String toString()
toString
in class LogisticBase
public java.lang.String getModelParameters()
public int assignIDs(int lastID)
public int assignLeafModelNumbers(int leafCounter)
public java.lang.String modelsToString()
public java.lang.String graph() throws java.lang.Exception
java.lang.Exception
- if something goes wrongpublic void cleanup()
cleanup
in class LogisticBase
public java.lang.String getRevision()
getRevision
in interface RevisionHandler
getRevision
in class LogisticBase