From 0f73fa5b8bd5441300d01f1106d5c71909e2ca97 Mon Sep 17 00:00:00 2001 From: Wenhao-H Date: Mon, 22 Oct 2018 11:29:05 -0400 Subject: [PATCH] small fixes --- src/mlpack/methods/cf/cf.hpp | 3 +++ src/mlpack/methods/cf/normalization/combined_normalization.hpp | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/src/mlpack/methods/cf/cf.hpp b/src/mlpack/methods/cf/cf.hpp index f44517eb94..5154c60d53 100644 --- a/src/mlpack/methods/cf/cf.hpp +++ b/src/mlpack/methods/cf/cf.hpp @@ -179,6 +179,9 @@ class CFType //! Get the cleaned data matrix. const arma::sp_mat& CleanedData() const { return cleanedData; } + //! Get the normalization object. + const NormalizationType& Normalization() const { return normalization; } + /** * Generates the given number of recommendations for all users. * diff --git a/src/mlpack/methods/cf/normalization/combined_normalization.hpp b/src/mlpack/methods/cf/normalization/combined_normalization.hpp index 3bd36a8451..6836707f14 100644 --- a/src/mlpack/methods/cf/normalization/combined_normalization.hpp +++ b/src/mlpack/methods/cf/normalization/combined_normalization.hpp @@ -93,7 +93,7 @@ class CombinedNormalization /** * Return normalizations tuple. */ - TupleType Normalizations() const + const TupleType& Normalizations() const { return normalizations; }