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) 2011 Heiko Strathmann 00008 * Copyright (C) 2011 Berlin Institute of Technology and Max-Planck-Society 00009 */ 00010 00011 #ifndef __STRATIFIEDCROSSVALIDATION_H_ 00012 #define __STRATIFIEDCROSSVALIDATION_H_ 00013 00014 #include <shogun/evaluation/SplittingStrategy.h> 00015 00016 namespace shogun 00017 { 00018 00019 class CLabels; 00020 00026 class CStratifiedCrossValidationSplitting: public CSplittingStrategy 00027 { 00028 public: 00030 CStratifiedCrossValidationSplitting(); 00031 00037 CStratifiedCrossValidationSplitting(CLabels* labels, index_t num_subsets); 00038 00040 inline virtual const char* get_name() const 00041 { 00042 return "StratifiedCrossValidation"; 00043 } 00044 protected: 00046 void build_subsets(); 00047 }; 00048 } 00049 00050 #endif /* __STRATIFIEDCROSSVALIDATION_H_ */