org.apache.spark.ml.regression

Class AFTCostFun

Object

org.apache.spark.ml.regression.AFTCostFun

All Implemented Interfaces:

breeze.linalg.ImmutableNumericOps<breeze.optimize.StochasticDiffFunction<breeze.linalg.DenseVector<Object>>>, breeze.linalg.NumericOps<breeze.optimize.StochasticDiffFunction<breeze.linalg.DenseVector<Object>>>, breeze.optimize.DiffFunction<breeze.linalg.DenseVector<Object>>, breeze.optimize.StochasticDiffFunction<breeze.linalg.DenseVector<Object>>, scala.Function1<breeze.linalg.DenseVector<Object>,Object>

public class AFTCostFun
extends Object
implements breeze.optimize.DiffFunction<breeze.linalg.DenseVector<Object>>

AFTCostFun implements Breeze's DiffFunction[T] for AFT cost. It returns the loss and gradient at a particular point (parameters). It's used in Breeze's convex optimization routines.

Nested Class Summary

Nested classes/interfaces inherited from interface breeze.linalg.NumericOps

breeze.linalg.NumericOps.Arrays$, breeze.linalg.NumericOps.ArraysLowPriority

Constructor Summary

Constructors

Constructor and Description
AFTCostFun(RDD<org.apache.spark.ml.regression.AFTPoint> data, boolean fitIntercept, Broadcast<double[]> bcFeaturesStd, int aggregationDepth)

Method Summary

Modifier and Type	Method and Description
scala.Tuple2<Object,breeze.linalg.DenseVector<Object>>	calculate(breeze.linalg.DenseVector<Object> parameters)

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface breeze.optimize.DiffFunction

$init$, cached, castOps, opAddDiffFunction, opDivDiffFunction, opDivLHSDiffFunction, opMulDiffFunction, opMulLHSDiffFunction, opSubDiffFunction, repr, throughLens, withL2Regularization, withL2Regularization

Methods inherited from interface breeze.optimize.StochasticDiffFunction

$init$, apply, gradientAt, valueAt

Methods inherited from interface scala.Function1

$init$, andThen, apply$mcDD$sp, apply$mcDF$sp, apply$mcDI$sp, apply$mcDJ$sp, apply$mcFD$sp, apply$mcFF$sp, apply$mcFI$sp, apply$mcFJ$sp, apply$mcID$sp, apply$mcIF$sp, apply$mcII$sp, apply$mcIJ$sp, apply$mcJD$sp, apply$mcJF$sp, apply$mcJI$sp, apply$mcJJ$sp, apply$mcVD$sp, apply$mcVF$sp, apply$mcVI$sp, apply$mcVJ$sp, apply$mcZD$sp, apply$mcZF$sp, apply$mcZI$sp, apply$mcZJ$sp, compose, toString

Methods inherited from interface breeze.linalg.NumericOps

$amp$eq, $bar$eq, $colon$amp$eq, $colon$bar$eq, $colon$div$eq, $colon$eq, $colon$minus$eq, $colon$percent$eq, $colon$plus$eq, $colon$times$eq, $colon$up$eq, $colon$up$up$eq, $div$eq, $greater$colon$eq, $greater$colon$greater, $init$, $less$colon$eq, $less$colon$less, $minus$eq, $percent$eq, $plus, $plus$eq, $times$eq, $up$up$eq, binaryUpdateOpFromDVVOp

Methods inherited from interface breeze.linalg.ImmutableNumericOps

$amp, $amp$colon$amp, $bar, $bar$colon$bar, $bslash, $colon$bang$eq, $colon$eq$eq, $div, $div$colon$div, $init$, $minus, $minus$colon$minus, $percent, $percent$colon$percent, $plus$colon$plus, $times, $times$colon$times, $up$colon$up, $up$up, $up$up$colon$up$up, dot, t, t, t, unary_$bang, unary_$minus

Constructor Detail

AFTCostFun

public AFTCostFun(RDD<org.apache.spark.ml.regression.AFTPoint> data,
                  boolean fitIntercept,
                  Broadcast<double[]> bcFeaturesStd,
                  int aggregationDepth)

Method Detail

calculate

public scala.Tuple2<Object,breeze.linalg.DenseVector<Object>> calculate(breeze.linalg.DenseVector<Object> parameters)

Specified by:

calculate in interface breeze.optimize.StochasticDiffFunction<breeze.linalg.DenseVector<Object>>