diff options
author | Samuel Fadel <samuelfadel@gmail.com> | 2015-10-05 14:20:47 -0300 |
---|---|---|
committer | Samuel Fadel <samuelfadel@gmail.com> | 2015-10-05 14:20:47 -0300 |
commit | 2007ebb8e431d1341583a03c22d440a7d0f79194 (patch) | |
tree | f79e6ff21b15d57ee85501cecfc9fe2bf2ef8c36 /effectivenessobserver.cpp | |
parent | e1e268ebbcd8f0d29b8faa187dde6981f82b027f (diff) |
Re-added effectiveness observer (still needs work).
Diffstat (limited to 'effectivenessobserver.cpp')
-rw-r--r-- | effectivenessobserver.cpp | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/effectivenessobserver.cpp b/effectivenessobserver.cpp index 305b3f8..61bc264 100644 --- a/effectivenessobserver.cpp +++ b/effectivenessobserver.cpp @@ -6,7 +6,7 @@ EffectiveInteractionEnforcer::EffectiveInteractionEnforcer(const arma::uvec &sam { } -void EffectiveInteractionEnforcer::setSelection(const arma::uvec &selection) +void EffectiveInteractionEnforcer::setSelection(const QSet<int> &selection) { m_selection = selection; } @@ -15,13 +15,15 @@ void EffectiveInteractionEnforcer::setMeasureDifference(const arma::vec &measure { m_measure = measure; - if (m_selection.n_elem == 0) { + if (m_selection.isEmpty()) { return; } - arma::uvec selectionIndices(m_selection); - for (auto it = selectionIndices.begin(); it != selectionIndices.end(); it++) { - *it = m_sampleIndices[*it]; + arma::uvec selectionIndices(m_selection.size()); + int i = 0; + for (auto it = m_selection.cbegin(); it != m_selection.cend(); it++) { + selectionIndices[i] = m_sampleIndices[*it]; + i++; } double diff = arma::mean(m_measure(selectionIndices)); |