RingBDStack/SocialED

A python library for social event detection

它解决了什么问题

SocialED 解决了因缺乏统一框架而导致社交事件检测 (SED) 模型部署与评估困难的问题。它提供了一个标准化的 Python 工具包,集成了众多算法与数据集,简化了从 Twitter 和 Facebook 等社交媒体平台识别与分析现实世界事件(如自然灾害或病毒式现象)的过程。

运作方式

该库基于 PyTorch 和 DGL 构建,实现了一个包含四个主要阶段的统一 API:

  1. Preprocess:处理图结构构建与标记化。
  2. Fit:训练检测算法并调整模型参数。
  3. Detection:使用训练好的模型从输入数据中识别事件。
  4. Evaluate:使用精确率、召回率与 F1-score 等指标衡量性能。

适用对象

它专为自然语言处理 (NLP) 与社交媒体分析领域的 AI 研究人员与从业者设计,他们需要一个一致的平台来基准测试不同的 SED 算法,或将其部署于危机管理、舆情监测与假新闻检测等应用中。

特色亮点

  • 丰富的算法库:包含 19 种检测算法,从经典方法 (LDA, Word2Vec) 到先进的 GNN (KPGNN, QSGNN) 与 PLM (BERT, SBERT) 皆有涵盖。
  • 多样化的数据集支持:集成了 15 个涵盖多种语言(英语、法语、阿拉伯语)与场景(一般事件、灾难性危机)的数据集。
  • 灵活的架构:提供模块化组件与一致的接口,实现不同模型间的无缝集成。
  • 硬件支持:支持 CPU 与 GPU,以进行高性能的训练与推理。

相关

  • 项目
  • 项目
  • 项目
  • 项目