MultiCox.h

#ifndef theplu_yat_regression_multi_cox
#define theplu_yat_regression_multi_cox

// $Id: MultiCox.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;
  class MatrixBase;
}

namespace regression {

  class MultiCox
  {
  public:
    MultiCox(void);

    MultiCox(const MultiCox& other);

    MultiCox(MultiCox&& other);

    ~MultiCox(void);

    MultiCox& operator=(const MultiCox& other);

    MultiCox& operator=(MultiCox&& other);

    void add(const utility::VectorBase& x, double time, bool event);

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

    void clear(void);

    void train(void);

    double b(size_t i) const;

    double z(size_t i) const;

    double p(size_t i) const;

    double hazard_ratio(size_t i) const;

    double hazard_ratio_lower_CI(size_t, double alpha=0.95) const;

    double hazard_ratio_upper_CI(size_t, double alpha=0.95) const;

    const utility::MatrixBase& covariance(void) const;
  private:
    class Impl;
    std::unique_ptr<Impl> pimpl_;
  };

}}}

#endif