Cox.h

#ifndef theplu_yat_regression_cox
#define theplu_yat_regression_cox

// $Id: Cox.h 4198 2022-08-19 06:26:14Z peter $

/*
  Copyright (C) 2022 Peter Johansson

  This file is part of the yat library, https://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 <https://www.gnu.org/licenses/>.
*/

#include <memory>
#include <vector>

namespace theplu {
namespace yat {
  namespace utility {
    class VectorBase;
  }
namespace regression {


  class Cox
  {
  public:
    Cox(void);

    Cox(const Cox& other);

    Cox(Cox&& other);

    ~Cox(void);

    Cox& operator=(const Cox& other);

    Cox& operator=(Cox&& other);

    void add(double x, double time, bool event);

    void add(const utility::VectorBase& x,
             const utility::VectorBase& time,
             const std::vector<char>& event);

    void clear(void);

    void train(void);

    double b(void) const;

    double z(void) const;

    double p(void) const;

    double hazard_ratio(void) const;

    double hazard_ratio_lower_CI(double alpha=0.95) const;

    double hazard_ratio_upper_CI(double alpha=0.95) const;
  private:
    class Impl;
    std::unique_ptr<Impl> pimpl_;
  };

}}}

#endif