Working with Datasets#
[1]:
import pymovements as pm
dataset = pm.datasets.ToyDataset(
root='data/',
download=True,
extract=True,
remove_finished=True,
)
/home/docs/checkouts/readthedocs.org/user_builds/pymovements/envs/v0.6.0/lib/python3.9/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html
from .autonotebook import tqdm as notebook_tqdm
Downloading http://github.com/aeye-lab/pymovements-toy-dataset/zipball/6cb5d663317bf418cec0c9abe1dde5085a8a8ebd/ to data/ToyDataset/downloads/pymovements-toy-dataset.zip
pymovements-toy-dataset.zip: 100%|██████████| 3.06M/3.06M [00:00<00:00, 24.6MB/s]
[2]:
subset = {
'text_id': [1, 2],
'page_id': 1,
}
dataset.load(subset=subset)
100%|██████████| 2/2 [00:00<00:00, 170.81it/s]
[3]:
dataset.fileinfo
[3]:
| text_id | page_id | filepath |
|---|---|---|
| i64 | i64 | str |
| 1 | 1 | "aeye-lab-pymov... |
| 2 | 1 | "aeye-lab-pymov... |
[4]:
dataset.gaze[0]
[4]:
<pymovements.gaze.gaze_dataframe.GazeDataFrame at 0x7f547b8be310>
[5]:
dataset.pix2deg()
dataset.gaze[0]
100%|██████████| 2/2 [00:00<00:00, 509.88it/s]
[5]:
<pymovements.gaze.gaze_dataframe.GazeDataFrame at 0x7f547b8be310>
[6]:
dataset.pos2vel(method='savitzky_golay', window_length=7, polyorder=2)
dataset.gaze[0]
100%|██████████| 2/2 [00:00<00:00, 309.01it/s]
[6]:
<pymovements.gaze.gaze_dataframe.GazeDataFrame at 0x7f547b8be310>
[7]:
dataset.detect_events(method=pm.events.microsaccades)
dataset.events[0]
2it [00:00, 66.92it/s]
[7]:
<pymovements.events.events.EventDataFrame at 0x7f54bc0773a0>
[8]:
dataset.detect_events(method=pm.events.ivt, velocity_threshold=45, minimum_duration=55)
dataset.events[0]
2it [00:00, 250.50it/s]
[8]:
<pymovements.events.events.EventDataFrame at 0x7f54bc0773a0>
[9]:
dataset.save()
100%|██████████| 2/2 [00:00<00:00, 1041.29it/s]
100%|██████████| 2/2 [00:00<00:00, 419.49it/s]
[10]:
del dataset
[11]:
dataset = pm.datasets.ToyDataset(
root='data/',
download=False,
extract=False,
remove_finished=False,
)
dataset.load(events=True, preprocessed=True, subset=subset)
display(dataset.gaze[0])
display(dataset.events[0])
100%|██████████| 2/2 [00:00<00:00, 886.28it/s]
100%|██████████| 2/2 [00:00<00:00, 848.62it/s]
<pymovements.gaze.gaze_dataframe.GazeDataFrame at 0x7f54bc0ca940>
<pymovements.events.events.EventDataFrame at 0x7f54bc05c040>