Skip to content
Geo Engine Docs

ml

Util functions for machine learning

MlModelConfig Objects

Section titled “MlModelConfig Objects”
@dataclass
class MlModelConfig()

Configuration for an ml model

register_ml_model

Section titled “register_ml_model”
def register_ml_model(onnx_model: ModelProto,
                      model_config: MlModelConfig,
                      upload_timeout: int = 3600,
                      register_timeout: int = 60) -> MlModelName

Uploads an onnx file and registers it as an ml model

model_dim_to_tensorshape

Section titled “model_dim_to_tensorshape”
def model_dim_to_tensorshape(model_dims)

Transform an ONNX dimension into a MlTensorShape3D

check_backend_constraints

Section titled “check_backend_constraints”
def check_backend_constraints(input_shape: MlTensorShape3D,
                              output_shape: MlTensorShape3D,
                              ge_tile_size=(512, 512))

Checks that the shapes match the constraintsof the backend

validate_model_config

Section titled “validate_model_config”
def validate_model_config(onnx_model: ModelProto, *,
                          input_type: RasterDataType,
                          output_type: RasterDataType,
                          input_shape: MlTensorShape3D,
                          out_shape: MlTensorShape3D)

Validates the model config. Raises an exception if the model config is invalid