source: trunk/yat/regression/OneDimensional.cc

// $Id: OneDimensional.cc 4207 2022-08-26 04:36:28Z peter $

/*
  Copyright (C) 2005 Peter Johansson
  Copyright (C) 2006, 2007, 2008 Jari Häkkinen, Peter Johansson
  Copyright (C) 2011, 2012, 2022 Peter Johansson

  This file is part of the yat library, http://dev.thep.lu.se/yat

  The yat library is free software; you can redistribute it and/or
  modify it under the terms of the GNU General Public License as
  published by the Free Software Foundation; either version 3 of the
  License, or (at your option) any later version.

  The yat library is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with yat. If not, see <http://www.gnu.org/licenses/>.
*/

#include <config.h>

#include "OneDimensional.h"

#include <ostream>

namespace theplu {
namespace yat {
namespace regression {

  OneDimensional::OneDimensional(void)
    : chisq_(0)
  {
  }

  OneDimensional::~OneDimensional(void)
  {
  }


  double OneDimensional::chisq(void) const
  {
    return chisq_;
  }


  double OneDimensional::prediction_error2(const double x) const
  {
    return s2()+standard_error2(x);
  }


  std::ostream& OneDimensional::print(std::ostream& os, const double min,
                                      double max, const unsigned int n) const
  {
    double dx;
    if (n>1)
      dx=(max-min)/(n-1);
    else{
      dx=1.0;
      max=min;
    }

    for ( double x=min; x<=max; x+=dx) {
      double y = predict(x);
      double y_err = sqrt(prediction_error2(x));
      os << x << "\t" << y << "\t" << y_err << "\n";
    }
    return os;
  }


  double OneDimensional::r2(void) const
  {
    return 1 - chisq()/ap_.y_averager().sum_xx_centered();
  }

  double OneDimensional::variance(void) const
  {
    return ap_.y_averager().variance();
  }

}}} // of namespaces regression, yat, and theplu