sksurv.linear_model.IPCRidge

class sksurv.linear_model.IPCRidge(alpha=1.0, fit_intercept=True, normalize=False, copy_X=True, max_iter=None, tol=0.001, solver='auto')[source]

Accelerated failure time model with inverse probability of censoring weights.

This model assumes a regression model of the form

\[\log y = \beta_0 + \mathbf{X} \beta + \epsilon\]

L2-shrinkage is applied to the coefficients \(\beta\) and each sample is weighted by the inverse probability of censoring to account for right censoring (under the assumption that censoring is independent of the features, i.e., random censoring).

See 1 for further description.

Parameters

alpha (float, optional, default: 1.0) – Small positive values of alpha improve the conditioning of the problem and reduce the variance of the estimates.

coef_

Weight vector.

Type

ndarray, shape = (n_features,)

References

1

W. Stute, “Consistent estimation under random censorship when covariables are present”, Journal of Multivariate Analysis, vol. 45, no. 1, pp. 89-103, 1993. doi:10.1006/jmva.1993.1028.

__init__(alpha=1.0, fit_intercept=True, normalize=False, copy_X=True, max_iter=None, tol=0.001, solver='auto')[source]

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__([alpha, fit_intercept, normalize, …])

Initialize self.

fit(X, y)

Build an accelerated failure time model.

predict(X)

Predict using the linear accelerated failure time model.

score(X, y[, sample_weight])

Returns the concordance index of the prediction.

fit(X, y)[source]

Build an accelerated failure time model.

Parameters
  • X (array-like, shape = (n_samples, n_features)) – Data matrix.

  • y (structured array, shape = (n_samples,)) – A structured array containing the binary event indicator as first field, and time of event or time of censoring as second field.

Returns

Return type

self

predict(X)[source]

Predict using the linear accelerated failure time model.

Parameters

X ({array-like, sparse matrix}, shape = (n_samples, n_features)) – Samples.

Returns

C – Returns predicted values on original scale (NOT log scale).

Return type

array, shape = (n_samples,)

score(X, y, sample_weight=None)[source]

Returns the concordance index of the prediction.

Parameters
  • X (array-like, shape = (n_samples, n_features)) – Test samples.

  • y (structured array, shape = (n_samples,)) – A structured array containing the binary event indicator as first field, and time of event or time of censoring as second field.

Returns

cindex – Estimated concordance index.

Return type

float