graph-of-words 0.1

A module to generate a graph of words

Graph of words

Create a graph of word representation from a text. Roughly based on Malliaros, F. D., & Skianis, K. (2015). Graph-Based Term Weighting for Text Categorization. 1473–1479. https://doi.org/10.1145/2808797.2808872

Example:

import graph
graph = graph.GraphOfWords(window_size=2)
graph.build_graph(
    'Roses are red. Violets are blue',
    # OR a sentences list['Roses  are  red.', 'Violets are blue'],
    remove_stopwords=False,
    workers=4
)

graph.display_graph()
graph.write_graph_edges('edges_list.txt')

Note that build_graph also accepts a list of sentences.

The following actions are performed:

1. Split the text into sentences if required
2. Slide a window across each sentence
3. Add a directed graph edge form the current word to the other ones within the window, the weight being the distance between the two words

Documentation

See the documentation.