Skip to main content

Module batch_predict

Module batch_predict 

Source
Expand description

Adaptive batch prediction for geospatial ML inference

This module provides:

§Algorithm

After each batch completes the batcher computes a rolling average over the last N observations and compares it to target_latency_ms:

  • Too slow (avg > target): shrink towards min_batch_size
  • Too fast (avg < target): grow towards max_batch_size

The magnitude of each adjustment is controlled by adaptation_rate.

Structs§

AdaptiveBatchConfig
Adaptive batch sizing configuration
AdaptiveBatcher
Adaptive batch size controller
PredictionRequest
Single prediction request carrying raw float tensors.
PredictionResult
Single prediction result produced by an inference run.