Havina is a Python library that can generate knowledge graphs triplets from an input text. Its implementation is based on the paper "Language models are open knowledge graphs" with some tweaks to improve performance. Havina can be used to evaluate the language comprehension of AI models or as a tool to extract triplets from text and build knowledge graphs.
Project description
Havina
Havina is a Python library that can generate knowledge graphs triplets from an input text. Its implementation is based on the paper "Language models are open knowledge graphs" with some tweaks to improve performance. Most notably, instead of summing the attention scores of each word in a relation, I am calculating their average.
The reasoning behind this change is that a simple sum of scores favors longer relations even if the extra words do not carry any relevant meaning.
Havina can be used to evaluate the language comprehension of AI models or as a tool to extract triplets from text and build knowledge graphs.
How to use it
pip install havina
After importing the GraphGenerator
class from havina, simply call the object
with the sentence to evaluate and an optional number of workers. Each worker will span
a different process and the algorithm will split the work between them.
For more information about the constructor parameters, check the Constructor parameters section.
from havina import GraphGenerator
text = 'John Lennon is a famous singer.'
generator = GraphGenerator(
top_k=4,
contiguous_token=False
)
triplets = generator(text, workers=1)
print(triplets)
The code above will print the following:
[
HeadTailRelations(
head=Entity(text='john lennon', wikidata_id=None),
tail=Entity(text='a famous singer', wikidata_id=None),
relations=['be'])
]
The returned type is a list of HeadTailReations
objects, each of which contains
the head and tail entities and the possible relations between them. Relations are
Python strings.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.