Struct ModelZooLoader 

pub struct ModelZooLoader { /* private fields */ }

Loads pretrained (or synthetic-seed) models from the ModelZoo registry.

Example

use oxirs_embed::model_zoo::{ModelZoo, ModelZooLoader};

let loader = ModelZooLoader::new(std::env::temp_dir()).accept_license();
// Would load a real checkpoint when source resolves to a real file:
// let model = loader.load("transe-fb15k237")?;

Implementations

impl ModelZooLoader

pub fn new(base_dir: impl Into<PathBuf>) -> Self

Create a loader that stores materialised checkpoints under base_dir.

pub fn with_zoo(zoo: &'static ModelZoo, base_dir: impl Into<PathBuf>) -> Self

Create a loader backed by a custom (non-global) ModelZoo.

pub fn accept_license(self) -> Self

Signal that the caller accepts all licenses (including restrictive ones such as CC-BY-NC-4.0).

pub fn load(&self, name: &str) -> Result<Box<dyn EmbeddingModel>, ModelZooError>

Load the model identified by name from the registry.

Returns a heap-allocated, type-erased EmbeddingModel.