(self)
| 46 | return roles |
| 47 | |
| 48 | def get_frames(self): |
| 49 | frames = Counter() |
| 50 | for sample in self.data: |
| 51 | g: Graph = sample['amr'] |
| 52 | for i in g.instances(): |
| 53 | t = i.target |
| 54 | cells = t.split('-') |
| 55 | if len(cells) == 2 and len(cells[1]) == 2 and cells[1].isdigit(): |
| 56 | frames[t] += 1 |
| 57 | return frames |
| 58 | |
| 59 | |
| 60 | class AMRPickleDataset(AMRDataset): |
no test coverage detected