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 __CLUSTERINGMUTUALINFORMATION_H__ 00012 #define __CLUSTERINGMUTUALINFORMATION_H__ 00013 00014 #include <shogun/evaluation/ClusteringEvaluation.h> 00015 00016 namespace shogun 00017 { 00018 00021 class CClusteringMutualInformation: public CClusteringEvaluation 00022 { 00023 public: 00025 CClusteringMutualInformation(): CClusteringEvaluation() {} 00026 00028 virtual ~CClusteringMutualInformation() {} 00029 00038 virtual float64_t evaluate(CLabels* predicted, CLabels* ground_truth); 00039 00041 virtual EEvaluationDirection get_evaluation_direction() 00042 { 00043 return ED_MINIMIZE; 00044 } 00045 00051 virtual const char* get_name() const 00052 { 00053 return "ClusteringMutualInformation"; 00054 } 00055 }; 00056 00057 } 00058 00059 #endif /* end of include guard: __CLUSTERINGMUTUALINFORMATION_H__ */