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 Sergey Lisitsyn 00008 * Copyright (C) 2011 Berlin Institute of Technology and Max-Planck-Society 00009 */ 00010 00011 #include <shogun/evaluation/MeanSquaredError.h> 00012 #include <shogun/labels/Labels.h> 00013 #include <shogun/labels/RegressionLabels.h> 00014 #include <shogun/mathematics/Math.h> 00015 00016 using namespace shogun; 00017 00018 float64_t CMeanSquaredError::evaluate(CLabels* predicted, CLabels* ground_truth) 00019 { 00020 ASSERT(predicted && ground_truth); 00021 ASSERT(predicted->get_num_labels() == ground_truth->get_num_labels()); 00022 ASSERT(predicted->get_label_type() == LT_REGRESSION); 00023 ASSERT(ground_truth->get_label_type() == LT_REGRESSION); 00024 int32_t length = predicted->get_num_labels(); 00025 float64_t mse = 0.0; 00026 for (int32_t i=0; i<length; i++) 00027 mse += CMath::sq(((CRegressionLabels*) predicted)->get_label(i) - ((CRegressionLabels*) ground_truth)->get_label(i)); 00028 mse /= length; 00029 return mse; 00030 }