00001 /* 00002 * This program is free software; you can redistribute it and/or modify 00003 * it under the terms of the GNU General Public License as published by 00004 * the Free Software Foundation; either version 3 of the License, or 00005 * (at your option) any later version. 00006 * 00007 * Written (W) 2012 Chiyuan Zhang 00008 * Copyright (C) 2012 Chiyuan Zhang 00009 */ 00010 00011 #ifndef ECOCFORESTENCODER_H__ 00012 #define ECOCFORESTENCODER_H__ 00013 00014 #include <shogun/multiclass/ecoc/ECOCDiscriminantEncoder.h> 00015 00016 namespace shogun 00017 { 00018 00029 class CECOCForestEncoder: public CECOCDiscriminantEncoder 00030 { 00031 public: 00033 CECOCForestEncoder(); 00034 00036 virtual ~CECOCForestEncoder() {} 00037 00039 virtual const char* get_name() const { return "ECOCForestEncoder"; } 00040 00042 int32_t get_num_trees() const { return m_num_trees; } 00043 00045 void set_num_trees(int32_t num_trees); 00046 }; 00047 00048 } /* shogun */ 00049 00050 #endif /* end of include guard: ECOCFORESTENCODER_H__ */ 00051
