source: trunk/test/poisson.cc @ 3615

// $Id: poisson.cc 3615 2017-02-06 02:31:36Z peter $

/*
  Copyright (C) 2017 Jari Häkkinen

  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 "Suite.h"

#include "yat/regression/Poisson.h"
#include "yat/utility/Matrix.h"
#include "yat/utility/Vector.h"

using namespace theplu::yat;

int main(int argc, char* argv[])
{
  test::Suite suite(argc, argv);

  regression::Poisson model;
  size_t n = 20;
  size_t p = 4;
  utility::Vector y(n);
  utility::Matrix X(n, p);
  model.fit(X, y);
  if (model.fit_parameters().size() != p+1) {
    suite.add(false);
    suite.err() << "error: size of fit_parameters: "
                << model.fit_parameters().size()
                << " expected " << p+1 << "\n";
  }
  model.predict(X.row_const_view(0));

  return suite.return_value();
}