使用自己的数据集跑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...
使用自己的数据集跑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...