SHOGUN  3.2.1
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
MeanSquaredError.cpp
Go to the documentation of this file.
1 /*
2  * This program is free software; you can redistribute it and/or modify
3  * it under the terms of the GNU General Public License as published by
4  * the Free Software Foundation; either version 3 of the License, or
5  * (at your option) any later version.
6  *
7  * Written (W) 2011 Sergey Lisitsyn
8  * Copyright (C) 2011 Berlin Institute of Technology and Max-Planck-Society
9  */
10 
12 #include <shogun/labels/Labels.h>
15 
16 using namespace shogun;
17 
19 {
20  ASSERT(predicted && ground_truth)
21  ASSERT(predicted->get_num_labels() == ground_truth->get_num_labels())
22  ASSERT(predicted->get_label_type() == LT_REGRESSION)
23  ASSERT(ground_truth->get_label_type() == LT_REGRESSION)
24  int32_t length = predicted->get_num_labels();
25  float64_t mse = 0.0;
26  for (int32_t i=0; i<length; i++)
27  mse += CMath::sq(((CRegressionLabels*) predicted)->get_label(i) - ((CRegressionLabels*) ground_truth)->get_label(i));
28  mse /= length;
29  return mse;
30 }

SHOGUN Machine Learning Toolbox - Documentation