如何更改 .flair 缓存文件夹位置 (https://github.com/flairNLP/flair)？

如何解决如何更改 .flair 缓存文件夹位置 (https://github.com/flairNLP/flair)？

这个问题是关于 NLP Python 模块 Flair (https://github.com/flairNLP/flair)。

模型的默认下载文件夹位于 flair 缓存 (~/.flair) 中。但是，在服务器上工作时，我更希望此下载发生在另一个位置，因为 /home 目录非常小。这是一个最小的工作示例。

from flair.data import Sentence
from flair.models import SequenceTagger

# make a sentence
sentence = Sentence('I love Berlin .')

# load the NER tagger
tagger = SequenceTagger.load('ner')

因此您可以看到设备上没有剩余空间并出现以下错误： (...) OSError: [Errno 28] No space left on device: '~/.flair/models/ner-english/tmp8js3y34i' (...)

解决方法

发布答案，以防有人来到这里并像我一样被卡住。

from pathlib import Path
import flair
#flair.cache_root = "/your/path/.flair" # DOES NOT WORK
flair.cache_root = Path("/your/path/.flair")# WORKS

,

您可以设置环境变量 FLAIR_CACHE_ROOT。通过查看根 init.py 文件找到。

在某些旧版本的 FlairNLP 中，这不起作用。 os.getenv 函数应该传入 Path:

cache_root = Path(os.getenv('FLAIR_CACHE_ROOT',Path(Path.home(),".flair")))