From 7ab195c462a377288cc3e30a557ceb66d4d98f51 Mon Sep 17 00:00:00 2001 From: Bert Peters Date: Fri, 6 Apr 2018 13:28:06 +0200 Subject: [PATCH] Hide interactions to non-active nodes. --- src/fmri/visualisations.cpp | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/fmri/visualisations.cpp b/src/fmri/visualisations.cpp index c38a306..40a8358 100644 --- a/src/fmri/visualisations.cpp +++ b/src/fmri/visualisations.cpp @@ -164,7 +164,9 @@ static Animation *getReLUAnimation(const fmri::LayerData &prevState, if (curState.shape().size() == 2) { EntryList results; for (auto i : Range(curState.numEntries())) { - results.emplace_back(changes[i], make_pair(i, i)); + if (curState.data()[i] > EPSILON) { + results.emplace_back(changes[i], make_pair(i, i)); + } } const auto maxValue = max_element(results.begin(), results.end())->first; @@ -196,7 +198,9 @@ static Animation *getNormalizingAnimation(const fmri::LayerData &prevState, cons EntryList entries; entries.reserve(scaling.size()); for (auto i : Range(scaling.size())) { - entries.emplace_back(scaling[i], make_pair(i, i)); + if (std::abs(curState.data()[i]) > EPSILON) { + entries.emplace_back(scaling[i], make_pair(i, i)); + } } auto max_val = *max_element(scaling.begin(), scaling.end());