Skip to content
Snippets Groups Projects
Commit 1089023c authored by Andi Gerken's avatar Andi Gerken
Browse files

Improved individual_iid individual speed performance

parent 8cf4c862
No related branches found
No related tags found
No related merge requests found
...@@ -599,7 +599,7 @@ def evaluate_tracks_distance( ...@@ -599,7 +599,7 @@ def evaluate_tracks_distance(
paths: Iterable[Union[str, Path]], paths: Iterable[Union[str, Path]],
labels: Iterable[str] = None, labels: Iterable[str] = None,
predicate=None, predicate=None,
max_timesteps=1000, max_timesteps=4000,
): ):
"""Evaluate the distances of two or more fish on the track. """Evaluate the distances of two or more fish on the track.
...@@ -623,7 +623,7 @@ def evaluate_tracks( ...@@ -623,7 +623,7 @@ def evaluate_tracks(
file_settings=None, file_settings=None,
lw_distances=False, lw_distances=False,
seed=42, seed=42,
max_timesteps=1000, max_timesteps=4000,
): ):
"""Evaluate the track. """Evaluate the track.
...@@ -676,14 +676,16 @@ def evaluate_tracks( ...@@ -676,14 +676,16 @@ def evaluate_tracks(
return fig return fig
def evaluate_individual_speed(**kwargs): def evaluate_individual_speed(speeds_turns_from_paths, **kwargs):
"""Evaluate the average speeds per individual""" """Evaluate the average speeds per individual"""
return evaluate_individuals(mode="speed", **kwargs) return evaluate_individuals(
speeds_turns_from_paths=speeds_turns_from_paths, mode="speed", **kwargs
)
def evaluate_individual_iid(**kwargs): def evaluate_individual_iid(poses_from_paths, **kwargs):
"""Evaluate the average iid per file""" """Evaluate the average iid per file"""
return evaluate_individuals(mode="iid", **kwargs) return evaluate_individuals(poses_from_paths=poses_from_paths, mode="iid", **kwargs)
def evaluate_individuals( def evaluate_individuals(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment