Project Icon

quantulum3

智能提取文本中的数量和单位

quantulum3是一个强大的Python库,专门用于从非结构化文本中智能提取数量、测量值及其单位。该库利用GloVe向量表示和维基百科页面进行单位消歧,支持290多个单位和75个实体,能够解析拼写数字、范围和不确定性。quantulum3还提供实体、单位和数量的导出导入功能,支持自定义单位和实体,具备多语言支持,并可通过训练分类器优化消歧效果。

quantulum3

Travis主分支构建状态 覆盖率状态 PyPI版本 PyPI - Python版本 PyPI - 状态

这是一个Python库,用于从非结构化文本中提取数量、测量值及其单位的信息。它能够基于相似单位在GloVe向量表示中的k近邻和它们的维基百科页面来消除歧义。

这是Marco Lagi原始版本grhawks分支recastrodiaz分支的Python 3兼容分支。 与最新版本sklearn的兼容性基于sohrabtowfighi的分支

用户指南

安装

pip install quantulum3

要安装用于使用或训练消歧分类器的依赖项,请使用

pip install quantulum3[classifier]

当解析器找到两个或更多匹配文本的单位时,会使用消歧分类器。

使用方法

>>> from quantulum3 import parser
>>> quants = parser.parse('我想要2升葡萄酒')
>>> quants
[Quantity(2, 'litre')]

Quantity类存储了它从中提取的原始文本表面形式,以及匹配的(开始,结束)位置:

>>> quants[0].surface
u'2升'
>>> quants[0].span
(7, 15)

value属性提供解析后的数值,unit.name属性提供解析后的单位名称:

>>> quants[0].value
2.0
>>> quants[0].unit.name
'litre'

还提供了一个内联解析器,可以在文本中嵌入解析后的数量(特别适用于调试):

>>> print parser.inline_parse('我想要2升葡萄酒')
我想要2升 {Quantity(2, "litre")} 葡萄酒

由于解析器也能解析无量纲数字, 这个库也可以用于简单的数字提取。

>>> print parser.parse('我想要两个')
[Quantity(2, 'dimensionless')]

单位和实体

所有单位(如)及其关联的实体(如体积)都与维基百科进行了协调:

>>> quants[0].unit
Unit(name="litre", entity=Entity("volume"), uri=https://en.wikipedia.org/wiki/Litre)

>>> quants[0].unit.entity
Entity(name="volume", uri=https://en.wikipedia.org/wiki/Volume)

这个库包含超过290个单位和75个实体。它还可以解析拼写出的数字、范围和不确定性:

>>> parser.parse('我想要一加仑啤酒')
[Quantity(1, 'gallon')]

>>> parser.parse('LHC以12.8–13.0 TeV的能量对质子束进行碰撞')
[Quantity(12.8, "teraelectronvolt"), Quantity(13, "teraelectronvolt")]

>>> quant = parser.parse('LHC以12.9±0.1 TeV的能量对质子束进行碰撞')
>>> quant[0].uncertainty
0.1

非标准单位通常没有维基百科页面。解析器仍会尝试根据它们的维度猜测其基础实体:

>>> parser.parse('声音以0.34 km/s的速度传播')[0].unit
Unit(name="kilometre per second", entity=Entity("speed"), uri=None)

导出/导入

实体、单位和数量可以导出为字典和JSON字符串:

>>> quant = parser.parse('我想要2升葡萄酒')
>>> quant[0].to_dict()
{'value': 2.0, 'unit': 'litre', "entity": "volume", 'surface': '2升', 'span': (7, 15), 'uncertainty': None, 'lang': 'en_US'}
>>> quant[0].to_json()
'{"value": 2.0, "unit": "litre", "entity": "volume", "surface": "2升", "span": [7, 15], "uncertainty": null, "lang": "en_US"}'

默认情况下,导出的字典中只包含单位/实体名称,但可以包含这些:

>>> quant = parser.parse('我想要2升葡萄酒')
>>> quant[0].to_dict(include_unit_dict=True, include_entity_dict=True)  # 同样的参数也适用于.to_json()
{'value': 2.0, 'unit': {'name': 'litre', 'surfaces': ['cubic decimetre', 'cubic decimeter', 'litre', 'liter'], 'entity': {'name': 'volume', 'dimensions': [{'base': 'length', 'power': 3}], 'uri': 'Volume'}, 'uri': 'Litre', 'symbols': ['l', 'L', 'ltr', 'ℓ'], 'dimensions': [{'base': 'decimetre', 'power': 3}], 'original_dimensions': [{'base': 'litre', 'power': 1, 'surface': '升'}], 'currency_code': None, 'lang': 'en_US'}, 'entity': 'volume', 'surface': '2升', 'span': (7, 15), 'uncertainty': None, 'lang': 'en_US'}

类似的导出语法也适用于导出Unit和Entity对象。

你可以从字典和JSON导入Entity、Unit和Quantity对象。这要求对象在导出时使用了include_unit_dict=Trueinclude_entity_dict=True(根据需要):

>>> quant_dict = quant[0].to_dict(include_unit_dict=True, include_entity_dict=True)
>>> quant = Quantity.from_dict(quant_dict)
>>> ent_json = "{'name': 'volume', 'dimensions': [{'base': 'length', 'power': 3}], 'uri': 'Volume'}"
>>> ent = Entity.from_json(ent_json)

消歧

如果解析器检测到歧义,一个基于歧义单位或实体的维基百科页面的分类器会尝试猜测正确的一个:

>>> parser.parse('我在这上面花了20英镑!')
[Quantity(20, "pound sterling")]

>>> parser.parse('它的重量不超过20磅')
[Quantity(20, "pound-mass")]

或者:

>>> text = '地球的平均密度约为5.5x10-3 kg/cm³'
>>> parser.parse(text)[0].unit.entity
Entity(name="density", uri=https://en.wikipedia.org/wiki/Density)

>>> text = '大气中O₂的含量为2.98e-4 kg/升'
>>> parser.parse(text)[0].unit.entity
Entity(name="concentration", uri=https://en.wikipedia.org/wiki/Concentration)

除此之外,分类器还根据所有单位表面形式在GloVe向量表示中的距离,对最相似的词进行训练。

口语版本

Quantulum类包含将它们转换为可说出单位的方法。

>>> parser.parse("给我10e9 GW,现在!")[0].to_spoken()
一百亿千兆瓦
>>> parser.inline_parse_and_expand("现在给我$1e10,还有1 TW和0.5 J!")
现在给我一百亿美元,还有一太瓦和零点五焦耳!

操作

虽然这个库内部不能操作数量,但有许多很好的选择:

扩展

训练分类器

如果你想自己训练分类器,你需要分类器的依赖项(参见安装)。

使用命令行中的quantulum3-training,脚本quantulum3/scripts/train.pyquantulum3.classifier中的train_classifier方法来训练分类器。

quantulum3-training --lang <语言> --data <训练文件路径.json> --output <输出文件路径.joblib>

你可以向训练命令传入多个训练文件。输出格式为joblib。

要使用你的自定义模型,将训练好的模型文件路径传递给解析器:

parser = Parser.parse(<文本>, classifier_path="模型路径.joblib")

示例训练文件可以在quantulum3/_lang/<语言>/train中找到。

如果你想创建新的或不同的similars.json,安装pymagnitude

要从向量词表示文件中提取最近邻,使用scripts/extract_vere.py。它会自动提取每个歧义单位可能表面形式的向量表示的k个最近邻。结果邻居存储在quantulum3/similars.json中,并自动包含在训练中。

提供的文件应该是.magnitude格式,因为其他格式会先在运行时转换为.magnitude文件。查看预格式化的Magnitude格式流行词嵌入Magnitude了解更多信息。

额外单位

可以向quantulum添加额外的实体和单位以进行解析。这些将被添加到默认的单位和实体中。以下代码为示例调用:

>>> from quantulum3.load import add_custom_unit, remove_custom_unit
>>> add_custom_unit(name="schlurp", surfaces=["slp"], entity="dimensionless")
>>> parser.parse("这个极其锋利的工具精确到0.5 slp")
[Quantity(0.5, "Unit(name="schlurp", entity=Entity("dimensionless"), uri=None)")]

add_custom_unit函数的关键字参数直接转换为要创建的单位的属性。

自定义单位和实体

可以加载完全自定义的单位和实体集。这可以通过向load_custom_units和load_custom_entities函数传递文件路径列表来实现。加载自定义单位和实体将替换通常加载的默认单位和实体。

推荐使用上下文管理器来加载数量:

>>> from quantulum3 import load, parser
>>> with load.CustomQuantities(["单位文件路径.json"], ["实体文件路径.json"]):
>>>     parser.parse("这个极其锋利的工具精确到0.5 slp")

[Quantity(0.5, "Unit(name="schlurp", entity=Entity("dimensionless"), uri=None)")]

>>> # 默认单位和实体再次被加载

但也可以手动加载自定义单位和实体:

>>> from quantulum3 import load, parser

>>> load.load_custom_units(["单位文件路径.json"])
>>> load.load_custom_entities(["实体文件路径.json"])
>>> parser.parse("这个极其锋利的工具精确到0.5 slp")

[Quantity(0.5, "Unit(name="schlurp", entity=Entity("dimensionless"), uri=None)")]

移除自定义单位和实体,加载默认单位和实体

load.reset_quantities()


有关单位和实体文件格式的更多信息,请参阅下面的开发者指南。

## 开发者指南

### 添加单位和实体

完整的单位列表请参见 *units.json*,完整的实体列表请参见 *entities.json*。添加单位的标准如下:

- 该单位有(或被重定向到)维基百科页面
- 该单位在普遍使用中(例如,不包括[前公制瑞典计量单位](https://en.wikipedia.org/wiki/Swedish_units_of_measurement#Length))。

扩展这两个文件以包含感兴趣的单位/实体很容易。以下是 *entities.json* 中一个条目的示例:

```json
"speed": {
    "dimensions": [{"base": "length", "power": 1}, {"base": "time", "power": -1}],
    "URI": "https://en.wikipedia.org/wiki/Speed"
}
  • 实体的名称是其键。名称必须是唯一的。
  • URI是该实体的维基百科页面的名称。(即 https://en.wikipedia.org/wiki/Speed => Speed)
  • dimensions是维度,一个字典列表,每个字典都有一个base(另一个实体的名称)和一个power(可以为负的整数)。

以下是 units.json 中一个条目的示例:

"metre per second": {
    "surfaces": ["metre per second", "meter per second"],
    "entity": "speed",
    "URI": "Metre_per_second",
    "dimensions": [{"base": "metre", "power": 1}, {"base": "second", "power": -1}],
    "symbols": ["mps"]
},
"year": {
    "surfaces": [ "year", "annum" ],
    "entity": "time",
    "URI": "Year",
    "dimensions": [],
    "symbols": [ "a", "y", "yr" ],
    "prefixes": [ "k", "M", "G", "T", "P", "E" ]
}
  • 单位的名称是其键。名称必须是唯一的。
  • URI遵循与 entities.json 中相同的方案
  • surfaces是指代该单位的字符串列表。该库会处理复数形式,无需指定。
  • entityentities.json 中一个实体的名称
  • dimensions遵循与 entities.json 中相同的模式,但base是另一个单位的名称,而不是另一个实体的名称。
  • symbols是该单位可能的符号和缩写列表。
  • prefixes是一个可选列表。它可以包含公制二进制前缀,并自动生成相应的单位。如果你想添加特定内容(如不同的表面),你需要为该前缀版本单独创建一个条目。

所有字段都区分大小写。

贡献

dev 构建:

Travis dev build state Coverage Status

如果你想贡献,请按以下步骤操作:

  1. 将此项目的分支克隆到你的工作空间
  2. 在开发文件夹的根目录运行 pip install -e .
  3. pip install pipenvpipenv shell
  4. 在项目文件夹内运行 pipenv install --dev
  5. 进行你的更改
  6. 从包的根目录运行 scripts/format.shscripts/build.py
  7. 使用 python3 setup.py test 测试你的更改 (可选,推送后会自动执行)
  8. 提交并推送你的更改后,创建一个拉取请求

语言支持

Travis dev build state Coverage Status

有一个用于语言支持的分支,名为 language_support。 通过检查 _lang 子目录中的 README 文件以及 新的 _lang.en_US 子模块中给出的函数和值, 应该能够创建自己的语言子模块。 新的语言模块应该自动被调用并可用, 既可以通过解析器函数中的 lang= 关键字参数使用, 也可以在自动单元测试中使用。

除了自己的语言子模块文件夹(即 _lang.de_DE)之外,不应该需要进行任何更改。如果在实现新语言时遇到问题,请不要犹豫,打开一个问题。

项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号