ShareBoost.h

浏览该文件的文档.

/*
 * 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 SHAREBOOST_H__
#define SHAREBOOST_H__

#include <shogun/lib/config.h>

#include <shogun/machine/LinearMulticlassMachine.h>
#include <shogun/multiclass/MulticlassOneVsRestStrategy.h>
#include <shogun/features/DenseFeatures.h>
#include <shogun/labels/MulticlassLabels.h>

namespace shogun
{

class CShareBoost: public CLinearMulticlassMachine
{
public:
    CShareBoost();

    CShareBoost(CDenseFeatures<float64_t> *features, CMulticlassLabels *labs, int32_t num_nonzero_feas);

    virtual ~CShareBoost() {}

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

    void set_num_nonzero_feas(int32_t n) { m_nonzero_feas = n; }

    int32_t get_num_nonzero_feas() const { return m_nonzero_feas; }

    void set_features(CFeatures *f);

    SGVector<int32_t> get_activeset();

    friend class ShareBoostOptimizer;
protected:

    virtual bool train_machine(CFeatures* data = NULL);

private:
    void init_sb_params(); 

    void compute_rho(); 
    int32_t choose_feature(); 
    void optimize_coefficients(); 
    void compute_pred(); 
    void compute_pred(const float64_t *W); 

    int32_t m_nonzero_feas; 
    SGVector<int32_t> m_activeset; 

    SGMatrix<float64_t> m_fea; 
    SGMatrix<float64_t> m_rho; 
    SGVector<float64_t> m_rho_norm; 
    SGMatrix<float64_t> m_pred; 
};

} /* shogun */

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