diff --git a/src/mlpack/methods/ann/loss_functions/log_cosh_loss.hpp b/src/mlpack/methods/ann/loss_functions/log_cosh_loss.hpp index 650c87a94d..788c2ef842 100644 --- a/src/mlpack/methods/ann/loss_functions/log_cosh_loss.hpp +++ b/src/mlpack/methods/ann/loss_functions/log_cosh_loss.hpp @@ -75,9 +75,9 @@ class LogCoshLoss OutputDataType& OutputParameter() { return outputParameter; } //! Get the value of hyperparameter a. - bool A() const { return a; } + double A() const { return a; } //! Modify the value of hyperparameter a. - bool& A() { return a; } + double& A() { return a; } /** * Serialize the loss function.