ClusteringMutualInformation.h

Go to the documentation of this file.
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__ */
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines

SHOGUN Machine Learning Toolbox - Documentation