com.intel.analytics.bigdl

optim

package optim

Visibility
  1. Public
  2. All

Type Members

  1. class AccuracyResult extends ValidationResult

    Represent an accuracy result.

  2. class Adadelta[T] extends OptimMethod[T]

    Adadelta implementation for SGD: http://arxiv.

  3. class Adagrad[T] extends OptimMethod[T]

    An implementation of Adagrad.

  4. class Adam[T] extends OptimMethod[T]

    An implementation of Adam http://arxiv.

  5. class Adamax[T] extends OptimMethod[T]

    An implementation of Adamax http://arxiv.

  6. class DistriOptimizer[T] extends Optimizer[T, MiniBatch[T]]

    The optimizer run on a distributed cluster.

  7. class DistriValidator[T] extends Validator[T, MiniBatch[T]]

    Validate model on a distributed cluster.

  8. class Evaluator[T] extends Serializable

    model evaluator

  9. class L1L2Regularizer[T] extends Regularizer[T]

    Apply both L1 and L2 regularization

  10. case class L1Regularizer[T](l1: Double)(implicit evidence$3: ClassTag[T], ev: TensorNumeric[T]) extends L1L2Regularizer[T] with Product with Serializable

    Apply L1 regularization

  11. case class L2Regularizer[T](l2: Double)(implicit evidence$4: ClassTag[T], ev: TensorNumeric[T]) extends L1L2Regularizer[T] with Product with Serializable

    Apply L2 regularization

  12. class LBFGS[T] extends OptimMethod[T]

    This implementation of L-BFGS relies on a user-provided line search function (state.

  13. trait LineSearch[T] extends AnyRef

    Line Search strategy

  14. class LocalOptimizer[T] extends Optimizer[T, MiniBatch[T]]

    Optimize a model on a single machine

  15. class LocalPredictor[T] extends Serializable

  16. class LocalValidator[T] extends Validator[T, MiniBatch[T]]

    Validate a model on a single machine Use given dataset with certain validation methods such as Top1Accuracy as an argument of its test method

  17. class Loss[T] extends ValidationMethod[T]

    This evaluation method is calculate loss of output with respect to target

  18. class LossResult extends ValidationResult

    Use loss as a validation result

  19. class MAE[T] extends ValidationMethod[T]

    This evaluation method is calculate mean absolute error of output with respect to target

  20. class Metrics extends Serializable

    Performance metrics for the training process.

  21. trait OptimMethod[T] extends Serializable

    Similar to torch Optim method, which is used to update the parameter

  22. abstract class Optimizer[T, D] extends AnyRef

    Optimizer is an abstract class which is used to train a model automatically with some certain optimization algorithms.

  23. class Predictor[T] extends Serializable

  24. class RMSprop[T] extends OptimMethod[T]

    An implementation of RMSprop

  25. trait Regularizer[T] extends Serializable

    It is a trait for all regularizers.

  26. class SGD[T] extends OptimMethod[T]

    A plain implementation of SGD

  27. class Top1Accuracy[T] extends ValidationMethod[T]

    Caculate the percentage that output's max probability index equals target

  28. class Top5Accuracy[T] extends ValidationMethod[T]

    Caculate the percentage that target in output's top5 probability indexes

  29. class TreeNNAccuracy[T] extends ValidationMethod[T]

    This is a metric to measure the accuracy of Tree Neural Network/Recursive Neural Network

  30. trait Trigger extends AnyRef

    A trigger specifies a timespot or several timespots during training, and a corresponding action will be taken when the timespot(s) is reached.

  31. trait ValidationMethod[T] extends Serializable

    A method defined to evaluate the model.

  32. trait ValidationResult extends Serializable

    A result that calculate the numeric value of a validation method.

  33. abstract class Validator[T, D] extends AnyRef

    Validator is an abstract class which is used to test a model automatically with some certain validation methods such as Top1Accuracy, as an argument of its test method.

Value Members

  1. object DistriOptimizer

  2. object DistriValidator

  3. object EvaluateMethods

  4. object Evaluator extends Serializable

  5. object L1L2Regularizer extends Serializable

  6. object LocalOptimizer

  7. object LocalPredictor extends Serializable

  8. object LocalValidator

  9. object OptimMethod extends Serializable

  10. object Optimizer

  11. object Predictor extends Serializable

  12. object SGD extends Serializable

  13. object Trigger

Deprecated Value Members

  1. object Validator

    Annotations
    @deprecated
    Deprecated

    (Since version 0.2.0) Validator(model, dataset) is deprecated. Please use model.evaluate instead

Ungrouped