TabPFN在MacOS系统运行问题解决方案——下载模型

TabPFN问题解决

论文地址：https://pmc.ncbi.nlm.nih.gov/articles/PMC11711098/#notes3

GitHub地址：https://github.com/PriorLabs/tabpfn

模型地址：https://zenodo.org/records/13981285

问题描述：

运行以下代码

pip install tabpfn

出现报错，下载不了模型

RuntimeError: Failed to download model to /Users/Library/Caches/tabpfn/tabpfn-v2-regressor.ckpt!

是因为国内网络问题，下载不了模型，此时我们需要下载相关的模型。

地址：https://zenodo.org/records/13981285

下载models.zip

下载到本地后，我们需要修改TabPFN-main中的代码，因为代码中的模型是自动下载的，我们需要手动修改成我们下载到本地的模型。

位置在TabPFN-main/src/tabpfn/base.py

需要修改的内容：

    download = False
    if isinstance(model_path, str) and model_path == "/yourpath/TabPFN-main/models/tabpfn-v2-regressor":
        model_path = None  # type: ignore

此时，上述报错解决了。

但是运行网页测试代码，发现报错

RuntimeError: Failed to download model to RuntimeError: Failed to download model to /Users/a1-6/Library/Caches/tabpfn/tabpfn-v2-regressor.ckpt!

没有找到相关的模型，我是按照报错建议。新建了目录，将文件移动到该路径下。

新建目录

mkdir -p /Users/a1-6/Library/Caches/tabpfn/

移动文件

mv ~/Documents/Code/python/missing/TabPFN-main/models/tabpfn-v2-regressor  /Users/a1-6/Library/Caches/tabpfn/

运行测试代码，成功。

原因：

可能代码中，针对macOS系统的models是通过以上路径加载的，我没有找到相关的代码，只能通过新建目录，移动文件实现运行。

 

———————————————————————————————————————————————————————————————————————————仔细查看了GitHub描述文档，是自己没有看清楚。

如下