Source code for h2o.model.models.coxph

# -*- encoding: utf-8 -*-
from h2o.model import ModelBase


class H2OCoxPHModel(ModelBase):

    def formula(self):
        """Survival formula."""
        return self._model_json["output"]["formula"] 
        
    def concordance(self):
        """Concordance"""
        return self._model_json["output"]["concordance"]

    def coefficients_table(self):
        """Coefficients table."""
        return self._model_json["output"]["coefficients_table"]
    
    def summary(self):
        """legacy behaviour as for some reason, CoxPH is formatting summary differently than other models"""
        return self._summary()

    def get_summary(self):
        output = self._model_json["output"]
        return """Call:
{formula}
{coefs}
Likelihood ratio test={lrt:f}
Concordance={concordance:f}
n={n:d}, number of events={tot_events:d}
""".format(formula=self.formula(),
           coefs=self.coefficients_table(),
           lrt=output["loglik_test"],
           concordance=self.concordance(),
           n=output['n'],
           tot_events=output["total_event"])


class H2OCoxPHMojoModel(ModelBase):
    pass