apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix minor issues for when solutions are extremely sparse.

Browse Source
This commit is contained in:
Ryan Curtin
2024-05-18 22:36:51 -04:00
parent fa94886582
commit 3da96a3e12
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/local_coordinate_coding/lcc_impl.hpp
+6 -3
View File
@@ -232,7 +232,7 @@ inline void LocalCoordinateCoding<MatType>::OptimizeDictionary(
arma::uvec atomReverseLookup(atoms);
for (size_t i = 0; i < activeAtoms.size(); ++i)
{
atomReverseLookup[i] = activeAtoms[i];
atomReverseLookup[activeAtoms[i]] = i;
}
codesPrime(arma::span::all, arma::span(0, data.n_cols - 1)) = activeCodes;
@@ -266,8 +266,11 @@ inline void LocalCoordinateCoding<MatType>::OptimizeDictionary(
}
}
wSquared.subvec(data.n_cols, wSquared.n_elem - 1) = lambda *
abs(wSquared.subvec(data.n_cols, wSquared.n_elem - 1));
if (adjacencies.n_elem > 0)
{
wSquared.subvec(data.n_cols, wSquared.n_elem - 1) = lambda *
abs(wSquared.subvec(data.n_cols, wSquared.n_elem - 1));
}
// Solve system.
if (nInactiveAtoms == 0)