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 RELAXEDTREEUTIL_H__ 00012 #define RELAXEDTREEUTIL_H__ 00013 00014 #include <shogun/machine/BaseMulticlassMachine.h> 00015 #include <shogun/features/DenseFeatures.h> 00016 #include <shogun/labels/MulticlassLabels.h> 00017 #include <shogun/lib/SGMatrix.h> 00018 00019 namespace shogun 00020 { 00021 00023 class RelaxedTreeUtil 00024 { 00025 public: 00026 00034 SGMatrix<float64_t> estimate_confusion_matrix(CBaseMulticlassMachine *machine, CFeatures *X, CMulticlassLabels *Y, int32_t num_classes); 00035 00042 void get_confusion_matrix(SGMatrix<float64_t> &conf_mat, CMulticlassLabels *gt, CMulticlassLabels *pred); 00043 }; 00044 00045 } /* shogun */ 00046 00047 #endif /* end of include guard: RELAXEDTREEUTIL_H__ */ 00048
