wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update examples/1_load_hugging_face_dataset.py 

Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
This commit is contained in:
Remi 2024-04-18 11:04:19 +02:00 committed by GitHub
parent b4079a85e6
commit bae0c216bc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/1_load_hugging_face_dataset.py
View File

@ -44,7 +44,7 @@ print(f"{hf_dataset=}")
print(f"{hf_dataset.features=}") print(f"{hf_dataset.features=}")
# display useful statistics about frames and episodes, which are sequences of frames from the same video # display useful statistics about frames and episodes, which are sequences of frames from the same video
print(f"number of samples/frames: {len(hf_dataset)=}") print(f"number of frames: {len(hf_dataset)=}")
print(f"number of episodes: {len(hf_dataset.unique('episode_id'))=}") print(f"number of episodes: {len(hf_dataset.unique('episode_id'))=}")
print(f"average number of frames per episode: {len(hf_dataset) / len(hf_dataset.unique('episode_id')):.3f}") print(f"average number of frames per episode: {len(hf_dataset) / len(hf_dataset.unique('episode_id')):.3f}")