Skip to main content
Do you like Artifex? Give it a ⭐ star on GitHub!

named_entity_recognition.load()

Load a model checkpoint to perform inference with, or to fine-tune it further. Only models that were trained with Artifex can be loaded.

Arguments

  • model_path
    str

    The local path to the model checkpoint to load. This should be a directory containing all the model files generated by Artifex, such as config.json and model.safetensors.
from artifex import Artifex

ner = Artifex().named_entity_recognition

ner.load("/path/to/model/checkpoint")
None