solver

Type Members

1. class MultiLayerPerceptron extends Solver[Vector, MultiLayerPerceptron, MultiLayerPerceptronModel]

   

   Simple multilayer perceptron implementing backpropagation.

2. class MultiLayerPerceptronModel extends SolverModel[Vector, MultiLayerPerceptronModel]

   

3. trait Pretrainer extends Solver[Vector, UnsupervisedPretrainingSolver, UnsupervisedPretrainingSolverModel]

   

   Pretrainer provides a way to train pre train networks before running fully backpropagation.

   Pretrainer provides a way to train pre train networks before running fully backpropagation. Pretrainer assumes pretrain layers are put continuously at the head of network.

4. abstract class Solver[FeaturesType, E <: Solver[FeaturesType, E, M], M <: SolverModel[FeaturesType, M]] extends Predictor[FeaturesType, E, M] with HasWeightCol

   

   Solver implements distributed training algorithm for deep learning models.

   Solver implements distributed training algorithm for deep learning models. Currently this class is doing Back propagation under data parallelism schema.

5. abstract class SolverModel[FeaturesType, M <: SolverModel[FeaturesType, M]] extends PredictionModel[FeaturesType, M]

   

6. class UnsupervisedPretrainingSolver extends Solver[Vector, UnsupervisedPretrainingSolver, UnsupervisedPretrainingSolverModel] with Pretrainer

   

7. class UnsupervisedPretrainingSolverModel extends SolverModel[Vector, UnsupervisedPretrainingSolverModel]