ConditionalProbabilityTree.h

Go to the documentation of this file.

/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * Written (W) 2012 Chiyuan Zhang
 * Copyright (C) 2012 Chiyuan Zhang
 */

#ifndef CONDITIONALPROBABILITYTREE_H__
#define CONDITIONALPROBABILITYTREE_H__

#include <map>

#include <shogun/lib/config.h>

#include <shogun/features/streaming/StreamingDenseFeatures.h>
#include <shogun/multiclass/tree/TreeMachine.h>
#include <shogun/multiclass/tree/ConditionalProbabilityTreeNodeData.h>

namespace shogun
{

class CConditionalProbabilityTree: public CTreeMachine<ConditionalProbabilityTreeNodeData>
{
public:
    CConditionalProbabilityTree(int32_t num_passes=1)
        :m_num_passes(num_passes), m_feats(NULL)
    {
    }

    virtual ~CConditionalProbabilityTree() { SG_UNREF(m_feats); }

    virtual const char* get_name() const { return "ConditionalProbabilityTree"; }

    void set_num_passes(int32_t num_passes)
    {
        m_num_passes = num_passes;
    }

    int32_t get_num_passes() const
    {
        return m_num_passes;
    }

    void set_features(CStreamingDenseFeatures<float32_t> *feats)
    {
        SG_REF(feats);
        SG_UNREF(m_feats);
        m_feats = feats;
    }

    virtual CMulticlassLabels* apply_multiclass(CFeatures* data=NULL);

    virtual int32_t apply_multiclass_example(SGVector<float32_t> ex);

    void print_tree();
protected:
    virtual bool train_require_labels() const { return false; }

    virtual bool train_machine(CFeatures* data);

    void train_example(SGVector<float32_t> ex, int32_t label);

    void train_path(SGVector<float32_t> ex, bnode_t *node);

    void train_node(SGVector<float32_t> ex, float64_t label, bnode_t *node);

    float64_t predict_node(SGVector<float32_t> ex, bnode_t *node);

    int32_t create_machine(SGVector<float32_t> ex);

    virtual bool which_subtree(bnode_t *node, SGVector<float32_t> ex)=0;

    void compute_conditional_probabilities(SGVector<float32_t> ex);

    float64_t accumulate_conditional_probability(bnode_t *leaf);

    int32_t m_num_passes; 
    std::map<int32_t, bnode_t*> m_leaves; 
    CStreamingDenseFeatures<float32_t> *m_feats; 
};

} /* shogun */

#endif /* end of include guard: CONDITIONALPROBABILITYTREE_H__ */