bert/research-project
Archived
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Break after reaching small interactions. Refs #5.

Browse Source
This commit is contained in:
Bert Peters
2018-03-27 15:20:53 +02:00
parent a237058560
commit a72bb7542b
2 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/fmri/visualisations.cpp
View File

@@ -113,14 +113,18 @@ static Animation *getFullyConnectedAnimation(const fmri::LayerData &prevState, c
}
const auto desiredSize = min(INTERACTION_LIMIT, numEntries);
auto idx = arg_nth_element(interactions.begin(), interactions.begin() + desiredSize, interactions.end(), [](auto a, auto b) {
return abs(a) > abs(b);
});
auto idx = arg_partial_sort(interactions.begin(), interactions.begin() + desiredSize, interactions.end(),
[](auto a, auto b) {
return abs(a) > abs(b);
});
EntryList result;
result.reserve(desiredSize);
const auto normalizer = getNodeNormalizer(prevState);
for (auto i : idx) {
if (abs(interactions[i]) < EPSILON){
break;
}
result.emplace_back(interactions[i], make_pair((i % shape[1]) / normalizer, i / shape[1]));
}