(self, path_vid)
| 357 | class TikTecDataset(Dataset): |
| 358 | |
| 359 | def __init__(self, path_vid): |
| 360 | self.data_complete = pd.read_json('./data/data.json',orient='records',dtype=False,lines=True) |
| 361 | |
| 362 | self.vid = [] |
| 363 | with open(f'./data/vids/{path_vid}', "r") as fr: |
| 364 | for line in fr.readlines(): |
| 365 | self.vid.append(line.strip()) |
| 366 | self.data = self.data_complete[self.data_complete['video_id'].isin(self.vid)] |
| 367 | |
| 368 | def __len__(self): |
| 369 | return self.data.shape[0] |
nothing calls this directly
no outgoing calls
no test coverage detected