12345qiupeng
/
gtsam
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add argmax method to TableDistribution

release/4.3a0
Varun Agrawal 2025-01-04 15:47:16 -05:00
parent 14f32544d2
commit 5a8a9425f9
2 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
gtsam/discrete/TableDistribution.cpp
View File

@ -121,6 +121,23 @@ DiscreteConditional::shared_ptr TableDistribution::max(
return std::make_shared<TableDistribution>(m);
}
/* ************************************************************************ */
uint64_t TableDistribution::argmax() const {
uint64_t maxIdx = 0;
double maxValue = 0.0;
Eigen::SparseVector<double> sparseTable = table_.sparseTable();
for (SparseIt it(sparseTable); it; ++it) {
if (it.value() > maxValue) {
maxIdx = it.index();
maxValue = it.value();
}
}
return maxIdx;
}
/* ****************************************************************************/
void TableDistribution::prune(size_t maxNrAssignments) {
table_ = table_.prune(maxNrAssignments);

7
gtsam/discrete/TableDistribution.h
View File

@ -141,6 +141,13 @@ class GTSAM_EXPORT TableDistribution : public DiscreteConditional {
virtual DiscreteConditional::shared_ptr max(
const Ordering& keys) const override;
/**
* @brief Return assignment that maximizes value.
*
* @return maximizing assignment for the variables.
*/
uint64_t argmax() const;
/// @}
/// @name Advanced Interface
/// @{