在Google Colab安装NeuralCoref失败求助及替代方案咨询

一、修复NeuralCoref安装问题 #

NeuralCoref 4.0实际仅兼容spaCy 2.x版本（官方文档兼容说明未及时更新），且Python 3.10的编译环境对其旧版源码支持不佳，按以下步骤操作可解决：

• 卸载当前spaCy 3.6.1：

  !pip uninstall -y spacy
  

• 安装与NeuralCoref兼容的spaCy版本（推荐2.3.5）：

  !pip install spacy==2.3.5
  

• 安装编译依赖（Colab默认缺少部分编译组件）：

  !apt-get install -y build-essential python3-dev
  

• 安装NeuralCoref：

  !pip install neuralcoref
  

• 下载spaCy英文模型并验证：

  !python -m spacy download en_core_web_sm
  

  测试代码：

  import spacy
  import neuralcoref
  
  nlp = spacy.load('en_core_web_sm')
  neuralcoref.add_to_pipe(nlp)
  doc = nlp("Alice loves her cat. She plays with it every day.")
  print(doc._.coref_clusters)
  

二、无需NeuralCoref的简易指代消解方案 #

如果不想降级spaCy，推荐以下两种轻量方案：

方案1：Coreferee（与spaCy 3.x原生集成） #

Coreferee是专为spaCy 3.x开发的指代消解库，安装和使用都很简便：

• 安装：

  !pip install coreferee
  !python -m coreferee install en
  

• 使用示例：

  import spacy
  import coreferee
  
  nlp = spacy.load('en_core_web_sm')
  nlp.add_pipe('coreferee')
  doc = nlp("Alice loves her cat. She plays with it every day.")
  print(doc._.coref_chains)
  

方案2：HuggingFace Transformers预训练模型 #

利用预训练的指代消解模型，无需依赖spaCy特定版本：

• 安装依赖：

  !pip install transformers torch
  

• 使用示例（以roberta-large-coref为例）：

  from transformers import pipeline
  
  coref_pipeline = pipeline("coreference-resolution", model="roberta-large-coref")
  result = coref_pipeline("Alice loves her cat. She plays with it every day.")
  print(result)
  

内容的提问来源于stack exchange，提问作者codak black