OSError: We couldn't connect to 'https://huggingface.co' to load this model, couldn't find it in the cached files and it looks like albert-base-v2_output is not the path to a directory containing a config.json file.
Checkout your internet connection or see how to run the library in offline mode at 'https://huggingface.co/docs/transformers/installation#offline-mode'.
由于无法命令行连接huggingface,采用本地路径的方式。但是一共有三种方法连接模型
load标准预训练模型:huggingface标准预训练模型(国内一般行不通)
model = ClassificationModel(
"roberta", "roberta-base"
)
load社区预训练模型:(国内一般也不行)
model = ClassificationModel(
"bert", "KB/bert-base-swedish-cased"
)
load本地预训练模型
outputs/best_model为本地保存模型
model = ClassificationModel(
"bert", "outputs/best_model"
)