Adapt CustomLayer.
This commit is contained in:
@@ -2,7 +2,7 @@
|
||||
* @file tests/custom_layer.hpp
|
||||
* @author Projyal Dev
|
||||
*
|
||||
* A simple custom layer mimicing SigmoidLayer for testing if custom
|
||||
* A simple custom layer mimicking SigmoidLayer for testing if custom
|
||||
* layers work.
|
||||
*
|
||||
* mlpack is free software; you may redistribute it and/or modify it under the
|
||||
@@ -20,16 +20,17 @@
|
||||
|
||||
namespace mlpack {
|
||||
namespace ann {
|
||||
/**
|
||||
* Standard Sigmoid layer.
|
||||
*/
|
||||
template <
|
||||
class ActivationFunction = LogisticFunction,
|
||||
typename InputDataType = arma::mat,
|
||||
typename OutputDataType = arma::mat
|
||||
>
|
||||
using CustomLayer = BaseLayer<
|
||||
ActivationFunction, InputDataType, OutputDataType>;
|
||||
|
||||
/**
|
||||
* Standard Sigmoid layer.
|
||||
*/
|
||||
template <
|
||||
class ActivationFunction = LogisticFunction,
|
||||
typename InputDataType = arma::mat,
|
||||
typename OutputDataType = arma::mat
|
||||
>
|
||||
using CustomLayer = BaseLayer<
|
||||
ActivationFunction, InputDataType, OutputDataType>;
|
||||
|
||||
} // namespace ann
|
||||
} // namespace mlpack
|
||||
|
||||
Reference in New Issue
Block a user