AFTSurvivalRegression (Spark 3.5.5 JavaDoc) (original) (raw)


public class AFTSurvivalRegression
extends Regressor<Vector,AFTSurvivalRegression,AFTSurvivalRegressionModel>
implements AFTSurvivalRegressionParams, DefaultParamsWritable, org.apache.spark.internal.Logging
Fit a parametric survival regression model named accelerated failure time (AFT) model (see Accelerated failure time model (Wikipedia)) based on the Weibull distribution of the survival time.
Since 3.1.0, it supports stacking instances into blocks and using GEMV for better performance. The block size will be 1.0 MB, if param maxBlockSizeInMB is set 0.0 by default.
See Also:
Serialized Form