# -*- encoding: utf-8 -*-
from h2o.model import ModelBase
[docs]class H2OCoxPHModel(ModelBase):
[docs] def concordance(self):
"""Concordance"""
return self._model_json["output"]["concordance"]
[docs] def coefficients_table(self):
"""Coefficients table."""
return self._model_json["output"]["coefficients_table"]
[docs] def summary(self):
"""legacy behaviour as for some reason, CoxPH is formatting summary differently than other models"""
return self._summary()
[docs] 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"])
[docs]class H2OCoxPHMojoModel(ModelBase):
pass