ECOCRandomDenseEncoder.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 ECOCRANDOMDENSEENCODER_H__
#define ECOCRANDOMDENSEENCODER_H__

#include <shogun/lib/config.h>

#include <shogun/multiclass/ecoc/ECOCEncoder.h>

namespace shogun
{

class CECOCRandomDenseEncoder: public CECOCEncoder
{
public:
    CECOCRandomDenseEncoder(int32_t maxiter=10000, int32_t codelen=0, float64_t pposone=0.5);

    virtual ~CECOCRandomDenseEncoder() {}

    void set_probability(float64_t pposone);

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

    virtual SGMatrix<int32_t> create_codebook(int32_t num_classes);

    int32_t get_default_code_length(int32_t num_classes) const
    {
        return static_cast<int32_t>(CMath::round(10 * CMath::log(num_classes)));
    }

protected:
    int32_t   m_maxiter; 
    int32_t   m_codelen; 
    float64_t m_pposone; 

private:
    bool check_probability(float64_t pposone)
    {
        if (pposone >= 0.999 || pposone <= 0.0001)
            return false;
        return true;
    }

    void init();
};

} // namespace shogun

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