vipr.plugins.inference.interfaces package¶
Submodules¶
vipr.plugins.inference.interfaces.data_loader module¶
vipr.plugins.inference.interfaces.model_builder module¶
- class vipr.plugins.inference.interfaces.model_builder.ModelBuilderInterface(**kw: Any)¶
Bases:
Interface- abstract build(package_dir, package)¶
Build and return an executable model from a model package directory.
- Parameters:
package_dir – Path to the unpacked model package directory
package – Parsed ModelPackage object
- Returns:
Executable model object
vipr.plugins.inference.interfaces.model_loader module¶
vipr.plugins.inference.interfaces.postprocessor module¶
Postprocessor interface.
This file defines the interface for all postprocessor implementations.
vipr.plugins.inference.interfaces.predictor module¶
- class vipr.plugins.inference.interfaces.predictor.PredictorInterface(**kw: Any)¶
Bases:
InterfaceInterface for predictor handlers with unified DataSet support.
- abstract predict(dataset, model: Any, params: dict[str, Any] | None = None) dict[str, Any]¶
Make predictions using the model with unified DataSet interface.
- Parameters:
dataset – DataSet with batch-first format (batch_size, n_points)
model – Model to use for prediction
params – Additional parameters for prediction control
- Returns:
dict with prediction results in batch format
Module contents¶
Inference-specific interface contracts.
- class vipr.plugins.inference.interfaces.ModelBuilderInterface(**kw: Any)¶
Bases:
Interface- abstract build(package_dir, package)¶
Build and return an executable model from a model package directory.
- Parameters:
package_dir – Path to the unpacked model package directory
package – Parsed ModelPackage object
- Returns:
Executable model object
- class vipr.plugins.inference.interfaces.ModelLoaderInterface(**kw: Any)¶
Bases:
Interface- load_model(**kwargs)¶
Loads a model of the given type.
- Parameters:
**kwargs – Additional parameters for model initialization
- Returns:
The loaded model
- class vipr.plugins.inference.interfaces.PostprocessorInterface(**kw: Any)¶
Bases:
InterfaceInterface for postprocessor implementations.
Postprocessors take prediction results and apply additional processing to refine or enhance the results.
- class vipr.plugins.inference.interfaces.PredictorInterface(**kw: Any)¶
Bases:
InterfaceInterface for predictor handlers with unified DataSet support.
- abstract predict(dataset, model: Any, params: dict[str, Any] | None = None) dict[str, Any]¶
Make predictions using the model with unified DataSet interface.
- Parameters:
dataset – DataSet with batch-first format (batch_size, n_points)
model – Model to use for prediction
params – Additional parameters for prediction control
- Returns:
dict with prediction results in batch format