Skip to content
Snippets Groups Projects
  • Alexander Ulanov's avatar
    26867ebc
    [SPARK-11262][ML] Unit test for gradient, loss layers, memory management for multilayer perceptron · 26867ebc
    Alexander Ulanov authored
    1.Implement LossFunction trait and implement squared error and cross entropy
    loss with it
    2.Implement unit test for gradient and loss
    3.Implement InPlace trait and in-place layer evaluation
    4.Refactor interface for ActivationFunction
    5.Update of Layer and LayerModel interfaces
    6.Fix random weights assignment
    7.Implement memory allocation by MLP model instead of individual layers
    
    These features decreased the memory usage and increased flexibility of
    internal API.
    
    Author: Alexander Ulanov <nashb@yandex.ru>
    Author: avulanov <avulanov@gmail.com>
    
    Closes #9229 from avulanov/mlp-refactoring.
    26867ebc
    History
    [SPARK-11262][ML] Unit test for gradient, loss layers, memory management for multilayer perceptron
    Alexander Ulanov authored
    1.Implement LossFunction trait and implement squared error and cross entropy
    loss with it
    2.Implement unit test for gradient and loss
    3.Implement InPlace trait and in-place layer evaluation
    4.Refactor interface for ActivationFunction
    5.Update of Layer and LayerModel interfaces
    6.Fix random weights assignment
    7.Implement memory allocation by MLP model instead of individual layers
    
    These features decreased the memory usage and increased flexibility of
    internal API.
    
    Author: Alexander Ulanov <nashb@yandex.ru>
    Author: avulanov <avulanov@gmail.com>
    
    Closes #9229 from avulanov/mlp-refactoring.