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; }