Skip to content

Commit

Permalink
kmeans
Browse files Browse the repository at this point in the history
  • Loading branch information
sronilsson committed Nov 26, 2024
1 parent 4d857e5 commit dede7b7
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 2 deletions.
2 changes: 1 addition & 1 deletion simba/data_processors/circling_detector.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ class CirclingDetector(ConfigReader):
.. note::
We pass the names of the left and right ears, as the method will use body-parts to compute the `nape` location of the animal.
We pass the names of the left ear, right ear, and nose, as the method will use these body-parts to compute the direction of the animal in degrees.
:param Union[str, os.PathLike] data_dir: Path to directory containing pose-estimated body-part data in CSV format.
:param Union[str, os.PathLike] config_path: Path to SimBA project config file.
Expand Down
6 changes: 5 additions & 1 deletion simba/data_processors/freezing_detector.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,14 +32,18 @@ class FreezingDetector(ConfigReader):
Freezing is detected as `absent` when not present.
.. note::
We pass the names of the left and right ears, as the method will use body-parts to compute the `nape` location of the animal.
:param Union[str, os.PathLike] data_dir: Path to directory containing pose-estimated body-part data in CSV format.
:param Union[str, os.PathLike] config_path: Path to SimBA project config file.
:param Optional[str] nose_name: The name of the pose-estimated nose body-part. Defaults to 'nose'.
:param Optional[str] left_ear_name: The name of the pose-estimated left ear body-part. Defaults to 'left_ear'.
:param Optional[str] right_ear_name: The name of the pose-estimated right ear body-part. Defaults to 'right_ear'.
:param Optional[str] tail_base_name: The name of the pose-estimated tail base body-part. Defaults to 'tail_base'.
:param Optional[int] time_window: The time window in preceding seconds in which to evaluate freezing. Default: 3.
:param Optional[int] movement_threshold: A movement threshold in millimeters per second.
:param Optional[int] movement_threshold: A movement threshold in millimeters per second. Defaults to 5.
:param Optional[Union[str, os.PathLike]] save_dir: Directory where to store the results. If None, then results are stored in the ``logs`` directory of the SimBA project.
:example:
Expand Down

0 comments on commit dede7b7

Please sign in to comment.