Skip to content
Snippets Groups Projects
Commit 838df1b3 authored by Oleksandr Maksymets's avatar Oleksandr Maksymets
Browse files

commented episode filtering

parent 8c6d7b05
Branches
No related tags found
No related merge requests found
...@@ -309,13 +309,9 @@ class EpisodeIterator(Iterator): ...@@ -309,13 +309,9 @@ class EpisodeIterator(Iterator):
# sample episodes # sample episodes
if num_episode_sample >= 0: if num_episode_sample >= 0:
np.random.seed(100) # np.random.seed(100)
print(f"np.random.state {np.random.get_state()[0]}") # print(f"np.random.state {np.random.get_state()[0]}")
episodes = [ # episodes = [episode for episode in episodes if episode.episode_id in ["7", "77", "774", "10"]]
episode
for episode in episodes
if episode.episode_id in ["7", "77", "774", "10"]
]
episodes = np.random.choice( episodes = np.random.choice(
episodes, num_episode_sample, replace=False episodes, num_episode_sample, replace=False
) )
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment