gov.sandia.cognition.learning.algorithm.tree.RegressionTree Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cognitive-foundry Show documentation

A single jar with all the Cognitive Foundry components.

There is a newer version: 4.0.1

/*
 * File:                RegressionTree.java
 * Authors:             Justin Basilico
 * Company:             Sandia National Laboratories
 * Project:             Cognitive Foundry
 *
 * Copyright November 29, 2007, Sandia Corporation.  Under the terms of Contract
 * DE-AC04-94AL85000, there is a non-exclusive license for use of this work by
 * or on behalf of the U.S. Government. Export of this program may require a
 * license from the United States Government. See CopyrightHistory.txt for
 * complete details.
 *
 */

package gov.sandia.cognition.learning.algorithm.tree;

import gov.sandia.cognition.learning.function.regression.Regressor;

/**
 * The {@code RegressionTree} class extends the {@code DecisionTree} class
 * to implement a decision tree that does regression.
 *
 * @param   The input type the regression tree evaluates.
 * @author Justin Basilico
 * @since  2.0
 */
public class RegressionTree
    extends DecisionTree
    implements Regressor
{
    /**
     * Creates a new instance of RegressionTree.
     */
    public RegressionTree()
    {
        this(null);
    }
    
    /**
     * Creates a new instance of RegressionTree.
     *
     * @param  rootNode The root node of the tree.
     */
    public RegressionTree(
        final DecisionTreeNode rootNode)
    {
        super(rootNode);
    }

    @Override
    public RegressionTree clone()
    {
        return (RegressionTree) super.clone();
    }
    
    @Override
    public double evaluateAsDouble(
        final InputType input)
    {
        return this.evaluate(input);
    }

}