thunlp / OpenNE

An Open-Source Package for Network Embedding (NE)
MIT License
1.68k stars 485 forks source link

使用自己的数据集跑LINE没有labelfile跑不了 #127

Closed luoweiqi7 closed 1 year ago

luoweiqi7 commented 1 year ago

使用自己的数据集跑LINE时,输入文件中只有edgefile,没有labelfile文件,程序就不能跑,请问一下如何解决? 以下是输入语句:python -m openne --model line --local-dataset --root-dir /home/lwq/OpenNE/data/huawei1 --edgefile edges.txt 报错结果为下: File "网络表示学习OpenNE\src\openne\tasks\tasks.py", line 34, in train self.split_dataset(graph) File "网络表示学习\OpenNE\src\openne\tasks\tasks.py", line 29, in split_dataset graph.get_split_data(train_percent=self.kwargs['clf_ratio'], validate_size=valsize, seed=seed) File "网络表示学习\OpenNE\src\openne\dataloaders\graph.py", line 313, in get_split_data validate_size=validate_size, seed=seed) File "网络表示学习\OpenNE\src\openne\dataloaders\graph.py", line 259, in _split X, Y = self.labels() File "网络表示学习\OpenNE\src\openne\dataloaders\graph.py", line 113, in labels Y.append(self.G.nodes[i]['label']) KeyError: 'label' Creating test set using 50.0% nodes as training set...