mindspore.dataset.text.GloVe
- class mindspore.dataset.text.GloVe[源代码]
用于将tokens映射到向量的GloVe对象。
- from_file(file_path, max_vectors=None)[源代码]
从文件构建CharNGram向量。
- 参数:
file_path (str) - 包含向量的文件路径。预训练向量集的格式必须是 glove.6B.*.txt 。
max_vectors (int,可选) - 用于限制加载的预训练向量的数量。 大多数预训练的向量集是按词频降序排序的。因此,如果内存不能存放整个向量集,或者由于其他原因不需要, 可以传递 max_vectors 限制加载数量。默认值:None,无限制。
- 返回:
GloVe, 根据文件构建的GloVe向量。
- 异常:
RuntimeError - file_path 参数所指向的文件非法或者包含的数据异常。
ValueError - max_vectors 参数值错误。
TypeError - max_vectors 参数不是整数类型。
样例:
>>> import mindspore.dataset.text as text >>> glove = text.GloVe.from_file("/path/to/glove/file", max_vectors=None)