ContiniousSpace: Speedup get_neighbors
by using KDTree
#1619
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Speedup the
get_neighbors()
function inContiniousSpace
by using a SciPy KDTree. It improves performance especially with a large number of agents. For small number of agents, the performance is roughly the same, without large regressions.I've run some benchmarks on the boid_flockers model, with different number of agents, measuring the total runtime in seconds.
Please review carefully, both changes in functionality/behaviour as performance.