场景
使用PyG导入MovieLens数据集:
dataset = MovieLens(root='data/MovieLens', model_name='all-MiniLM-L6-v2')
print(dataset)
报错信息
F:\Anaconda3\python.exe ""
Downloading https://files.grouplens.org/datasets/movielens/ml-latest-small.zip
Extracting data\MovieLens\raw\ml-latest-small.zip
Processing...
Traceback (most recent call last):
File "", line 17, in <module>
dataset = MovieLens(root='data/MovieLens', model_name='all-MiniLM-L6-v2')
File "F:\Anaconda3\lib\site-packages\torch_geometric\datasets\movie_lens.py", line 40, in __init__
super().__init__(root, transform, pre_transform)
File "F:\Anaconda3\lib\site-packages\torch_geometric\data\in_memory_dataset.py", line 57, in __init__
super().__init__(root, transform, pre_transform, pre_filter)
File "F:\Anaconda3\lib\site-packages\torch_geometric\data\dataset.py", line 88, in __init__
self._process()
File "F:\Anaconda3\lib\site-packages\torch_geometric\data\dataset.py", line 171, in _process
self.process()
File "F:\Anaconda3\lib\site-packages\torch_geometric\datasets\movie_lens.py", line 61, in process
from sentence_transformers import SentenceTransformer
ModuleNotFoundError: No module named 'sentence_transformers'
Process finished with exit code 1
解决
安装sentence-transformers:
pip install -U sentence-transformers -i https://pypi.douban.com/simple