Skip to main content

A library for match labels of thesaurus concepts to text and assigning scores to found occurrences.

Project description

stwfsapy

Build Status codecov

About

This library provides functionality to find the labels of SKOS thesaurus concepts in (short) text. It is a reimplementation in Python of stwfsa combined with the concept scoring from [1]. A deterministic finite automaton is constructed from the labels of the thesaurus concepts to perform the matching. In addition, a classifier is trained to score the matched occurrences of the concepts.

Data Requirements

The construction for the automaton requires a SKOS thesaurus represented as a rdflib Graph. Concepts should be related to labels by skos:prefLabel or skos:altLabel. Concepts have to be identifiable by rdf:type. The training of the predictor requires labeled short text. Each training sample should be annotated with one or more concepts from the thesaurus. Currently the algorithm does not yield satisfactory results for longer texts, i.e., you should work with titles and keywords only, or possibly with abstracts but not with fulltexts.

Usage

Create predictor

First load your thesaurus.

from rdflib import Graph

g = Graph()
g.load('/path/to/your/thesaurus')

First, define the type URI for descriptors. If your thesaurus has a hierarchical structure that includes groups, you can optionally specify the type URI for sub-thesauri. In this case you should also specify the relationship that relates sub-thesauri to concepts. Furthermore you can indicate whether the thesaurus relation is a specialisation. For the STW this would be

descriptor_type_uri = 'http://zbw.eu/namespaces/zbw-extensions/Descriptor'
thsys_type_uri = 'http://zbw.eu/namespaces/zbw-extensions/Thsys'
thesaurus_relation_type_uri = 'http://www.w3.org/2004/02/skos/core#broader'
is_specialisation = False

Create the predictor

from stwfsapy.predictor import StwfsapyPredictor
p = StwfsapyPredictor(
    g,
    descriptor_type_uri,
    thsys_type_uri,
    thesaurus_relation_type_uri,
    is_specialisation,
    langs={'en'},
    simple_english_plural_rules=True)

The next step assumes you have loaded your texts into a list X and your labels in a list of lists y, such that for all indices 0 <= i < len(X). The list at y[i] contains the URIs to the correct concepts for X[i]. The concepts should be given by their URI. Then you can train the classifier:

p.fit(X, y)

Afterwards you can get the predicted concepts and scores:

p.suggest_proba(['one input text', 'A completely different input text.'])

Alternatively you can get a sparse matrix of scores by calling

p.predict_proba(['one input text', 'Another input text.'])

The indices of the concepts are stored in p.concept_map_.

Options

Input Type

The StwfsapyPredictor class has an option input that allows it to handle different types of inputs in the feature argument X of transform and fit methods.

  • "content" expects string input. This is the default.
  • "file" expects python file handles.
  • "filename" expects paths to files.

Text Vectorizer

StwfsapyPredictor can optionally use TFIDF features of the input texts to score the matches found by the finite state automaton. However this uses a lot of memory. Therefore it is disabled by default.

Save Model

A trained predictor p can be stored by calling p.store('/path/to/storage/location'). Afterwards it can be loaded as follows:

from stwfsapy.predictor import StwfsapyPredictor

StwfsapyPredictor.load('/path/to/storage/location')

References

[1] Toepfer, Martin, and Christin Seifert. "Content-based quality estimation for automatic subject indexing of short texts under precision and recall constraints." International Conference on Theory and Practice of Digital Libraries. Springer, Cham, 2018.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

stwfsapy-0.2.1.tar.gz (21.5 kB view hashes)

Uploaded Source

Built Distribution

stwfsapy-0.2.1-py3-none-any.whl (37.0 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page