# -----------------------------------------------------------------------------
# MIT License
#
# Copyright (c) 2024 Ontolearn Team
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
# -----------------------------------------------------------------------------
"""Model adapters."""
import inspect
import json
import logging
import re
from typing import TypeVar, List, Optional, Union
from owlapy.class_expression import OWLClassExpression
from owlapy.iri import IRI
from owlapy.owl_axiom import OWLAxiom
from owlapy.owl_individual import OWLNamedIndividual
from owlapy.abstracts import AbstractOWLReasoner
from ontolearn.abstracts import AbstractNode
from ontolearn.base_concept_learner import BaseConceptLearner
from .learners import CELOE, OCEL
from ontolearn.concept_learner import EvoLearner, NCES
from ontolearn.ea_algorithms import EASimple
from ontolearn.ea_initialization import EARandomWalkInitialization, EARandomInitialization, RandomInitMethod
from ontolearn.fitness_functions import LinearPressureFitness
from ontolearn.heuristics import CELOEHeuristic, OCELHeuristic
from ontolearn.knowledge_base import KnowledgeBase
from ontolearn.learning_problem import PosNegLPStandard
from ontolearn.refinement_operators import ModifiedCELOERefinement
from ontolearn.metrics import Accuracy, F1, Recall, Precision, WeightedAccuracy
from ontolearn.triple_store import TripleStoreKnowledgeBase
from ontolearn.value_splitter import BinningValueSplitter, EntropyValueSplitter
logger = logging.getLogger(__name__)
metrics = {'f1': F1,
'accuracy': Accuracy,
'recall': Recall,
'precision': Precision,
'weighted_accuracy': WeightedAccuracy
}
models = {'celoe': CELOE,
'ocel': OCEL,
'evolearner': EvoLearner,
'nces': NCES}
heuristics = {'celoe': CELOEHeuristic,
'ocel': OCELHeuristic}
[docs]
def compute_quality(KB, solution, pos, neg, qulaity_func="f1"): # pragma: no cover
func = metrics[qulaity_func]().score2
instances = set(KB.individuals(solution))
if isinstance(list(pos)[0], str):
instances = {ind.str.split("/")[-1] for ind in instances}
tp = len(pos.intersection(instances))
fn = len(pos.difference(instances))
fp = len(neg.intersection(instances))
tn = len(neg.difference(instances))
return func(tp=tp, fn=fn, fp=fp, tn=tn)[-1]
def _get_matching_opts(_Type, optargs, kwargs, *, prefix=None): # pragma: no cover
"""Find the keys in kwargs that are parameters of _Type.
If prefix is specified, the keys in kwargs need to be prefixed with prefix_.
"""
opts = {}
if prefix is None:
def p(s):
return s
else:
def p(s):
return prefix + "_" + s
sig = set()
sig.update(inspect.signature(_Type).parameters.keys())
sig.difference_update({'args', 'kwds'})
try:
sig.update(inspect.signature(_Type.__init__).parameters.keys())
sig.discard('self')
except AttributeError:
pass
for opt in sig:
if p(opt) in kwargs:
opts[opt] = kwargs.pop(p(opt))
elif transform_string(p(opt)) in kwargs:
opts[opt] = kwargs.pop(transform_string(p(opt)))
elif opt in optargs:
opts[opt] = optargs[opt]
return opts
_N = TypeVar('_N', bound=AbstractNode) #:
[docs]
class Trainer: # pragma: no cover
def __init__(self, learner: BaseConceptLearner, reasoner: AbstractOWLReasoner):
"""
A class to disentangle the learner from its training.
Args:
learner: The concept learner.
reasoner: The reasoner to use (should have the same ontology as the `kb` argument of the learner).
"""
assert reasoner.get_root_ontology().get_ontology_id().get_ontology_iri().as_str() == \
learner.kb.ontology.get_ontology_id().get_ontology_iri().as_str(), "New reasoner does not have " + \
"the same ontology as the learner!"
learner.reasoner = reasoner
self.learner = learner
self.reasoner = reasoner
[docs]
def fit(self, *args, **kwargs):
"""Run the concept learning algorithm according to its configuration.
Once finished, the results can be queried with the `best_hypotheses` function."""
self.learner.fit(*args, **kwargs)
[docs]
def best_hypotheses(self, n):
"""Get the current best found hypotheses according to the quality.
Args:
n: Maximum number of results.
Returns:
Iterable with hypotheses in form of search tree nodes.
"""
return self.learner.best_hypotheses(n)
[docs]
def predict(self, individuals: List[OWLNamedIndividual],
hypotheses: Optional[List[Union[_N, OWLClassExpression]]] = None,
axioms: Optional[List[OWLAxiom]] = None, n: int = 10):
"""Creates a binary data frame showing for each individual whether it is entailed in the given hypotheses
(class expressions). The individuals do not have to be in the ontology/knowledge base yet. In that case,
axioms describing these individuals must be provided.
The state of the knowledge base/ontology is not changed, any provided axioms will be removed again.
Args:
individuals: A list of individuals/instances.
hypotheses: (Optional) A list of search tree nodes or class expressions. If not provided, the
current :func:`BaseConceptLearner.best_hypothesis` of the concept learner are used.
axioms: (Optional) A list of axioms that are not in the current knowledge base/ontology.
If the individual list contains individuals that are not in the ontology yet, axioms
describing these individuals must be provided. The argument can also be used to add
arbitrary axioms to the ontology for the prediction.
n: Integer denoting number of ALC concepts to extract from search tree if hypotheses=None.
Returns:
Pandas data frame with dimensions |individuals|*|hypotheses| indicating for each individual and each
hypothesis whether the individual is entailed in the hypothesis.
"""
return self.learner.predict(individuals, hypotheses, axioms, n)
[docs]
def save_best_hypothesis(self, n: int = 10, path: str = 'Predictions', rdf_format: str = 'rdfxml') -> None:
"""Serialise the best hypotheses to a file.
Args:
n: Maximum number of hypotheses to save.
path: Filename base (extension will be added automatically).
rdf_format: Serialisation format. currently supported: "rdfxml".
"""
self.learner.save_best_hypothesis(n, path, rdf_format)
[docs]
def execute(args): # pragma: no cover
args_d = args.__dict__
learner_type = models[args.model]
optargs = {}
if args.sparql_endpoint:
kb = TripleStoreKnowledgeBase(args.sparql_endpoint)
else:
kb = KnowledgeBase(path=args.knowledge_base_path)
with open(args.path_learning_problem) as json_file:
examples = json.load(json_file)
pos = set(map(OWLNamedIndividual, map(IRI.create, set(examples['positive_examples']))))
neg = set(map(OWLNamedIndividual, map(IRI.create, set(examples['negative_examples']))))
lp = PosNegLPStandard(pos=pos, neg=neg)
if args.model in ["celoe", "ocel"]:
heur_func = heuristics[args.model](**_get_matching_opts(heuristics[args.model], {}, args_d))
refinement_op = ModifiedCELOERefinement(**_get_matching_opts(ModifiedCELOERefinement,
{"knowledge_base": kb,
"value_splitter": BinningValueSplitter(args.max_nr_splits)},
args_d))
optargs = {"knowledge_base": kb,
"quality_func": metrics[args.quality_metric](),
"heuristic_func": heur_func,
"refinement_operator": refinement_op}
elif args.model == "evolearner":
fit_func = LinearPressureFitness(**_get_matching_opts(LinearPressureFitness, {}, args_d))
init_rw_method = EARandomWalkInitialization(**_get_matching_opts(EARandomWalkInitialization, {}, args_d))
algorithm = EASimple(**_get_matching_opts(EASimple, {}, args_d))
mut_uniform_gen = EARandomInitialization(**_get_matching_opts(
EARandomInitialization, {"method": getattr(RandomInitMethod, args.init_method_type)}, args_d))
value_splitter = EntropyValueSplitter(**_get_matching_opts(EntropyValueSplitter, {}, args_d))
optargs = {"knowledge_base": kb,
"quality_func": metrics[args.quality_metric](),
"fitness_func": fit_func,
"init_method": init_rw_method,
"algorithm": algorithm,
"mut_uniform_gen": mut_uniform_gen,
"value_splitter": value_splitter}
# elif args.model == "drill":
# optargs = {"knowledge_base": kb,
# "quality_func": metrics[args.quality_metric]()}
model = learner_type(**_get_matching_opts(learner_type, optargs, args_d))
if args.model in ["celoe", "evolearner", "ocel"]:
trainer = Trainer(model, kb.reasoner)
trainer.fit(lp)
print(trainer.best_hypotheses(1))
if args.save:
trainer.save_best_hypothesis()
elif args.model in ["nces"]:
hypothesis = model.fit(pos, neg) # This will also print the prediction
# @TODO:CD: model.fit() should return a train model itself, not predictions
report = f"Quality: {compute_quality(kb, hypothesis, pos, neg, args.quality_metric)} \nIndividuals: " + \
f"{kb.individuals_count(hypothesis)}"
print(report)