Youtube social network dataset. Splits: graph, label
| 446 | |
| 447 | |
| 448 | class Youtube(Dataset): |
| 449 | """ |
| 450 | Youtube social network dataset. |
| 451 | |
| 452 | Splits: |
| 453 | graph, label |
| 454 | """ |
| 455 | def __init__(self): |
| 456 | super(Youtube, self).__init__( |
| 457 | "youtube", |
| 458 | urls={ |
| 459 | "graph": "http://socialnetworks.mpi-sws.mpg.de/data/youtube-links.txt.gz", |
| 460 | "label": "http://socialnetworks.mpi-sws.mpg.de/data/youtube-groupmemberships.txt.gz" |
| 461 | } |
| 462 | ) |
| 463 | |
| 464 | def label_preprocess(self, raw_file, save_file): |
| 465 | self.top_k_label(raw_file, save_file, k=47) |
| 466 | |
| 467 | |
| 468 | class Flickr(Dataset): |