cgboost.h

Go to the documentation of this file.

// -*- C++ -*-
#ifndef __LEMGA_AGGREGATING_CGBOOST_H__
#define __LEMGA_AGGREGATING_CGBOOST_H__

#include "boosting.h"

namespace lemga {

struct _boost_cg;

class CGBoost : public Boosting {
    std::vector<std::vector<REAL> > all_wgts;
    friend struct _boost_cg;

protected:
    /* only valid within training */
    std::vector<REAL> ncd,  
        cgd;  

public:
    explicit CGBoost (bool cvx = false, const cost::Cost& c = cost::_cost)
        : Boosting(cvx, c) {}
    explicit CGBoost (std::istream& is) { is >> *this; }

    virtual const id_t& id () const;
    virtual CGBoost* create () const { return new CGBoost(); }
    virtual CGBoost* clone () const { return new CGBoost(*this); }

    virtual bool set_aggregation_size (UINT);
    virtual void initialize ();
    virtual REAL train ();
    virtual REAL train_gd ();

protected:
    std::vector<REAL> cur_err;

    virtual REAL linear_weight (const DataWgt&, const LearnModel&);
    virtual void linear_smpwgt (DataWgt&);

    virtual bool serialize (std::ostream&, ver_list&) const;
    virtual bool unserialize (std::istream&, ver_list&,
                              const id_t& = empty_id);
};

struct _boost_cg : public _boost_gd {
    CGBoost* cg;
    _boost_cg (CGBoost* pc) : _boost_gd(pc), cg(pc) {}

    void set_weight (const Boosting::BoostWgt& bw) const {
        _boost_gd::set_weight(bw);
        assert(cg->n_in_agg == bw.size() && cg->n_in_agg == cg->lm_wgt.size());

        // save weights to all_wgts
        if (cg->n_in_agg == 0) return;
        const UINT n = cg->n_in_agg - 1;
        if (n < cg->all_wgts.size())
            cg->all_wgts[n] = cg->lm_wgt;
        else {
            assert(n == cg->all_wgts.size()); // allow size inc <= 1
            cg->all_wgts.push_back(cg->lm_wgt);
        }
    }
};

} // namespace lemga

#ifdef  __CGBOOST_H__
#warning "This header file may conflict with another `cgboost.h' file."
#endif
#define __CGBOOST_H__
#endif

---