snowflake.ml.model.batch_inference

Classes

BatchInferenceTask(name, *, model_version, ...)

A DAGTask that runs a batch inference job for a registered model version.

ColumnHandlingOptions

Options for handling specific columns during run_batch for file I/O.

EngineOptions(*[, engine, engine_args_override])

inference.engine_options sub-block of the job specification.

FileEncoding(value[, names, module, ...])

The encoding of the file content that will be passed to the custom model.

ImageBuildSpec(*[, image_repo, force_rebuild])

Image-build block of the batch inference job specification.

InferenceSpec(*[, num_workers, ...])

Inference block of the batch inference job specification.

InputFormat(value[, names, module, ...])

The format of the input column data.

InputSpec(*[, params, column_handling, ...])

Input block of the batch inference job specification.

OutputSpec(*, stage_location[, mode])

Output block of the batch inference job specification.

ResourcesSpec(*[, cpu_requests, ...])

Resources block of the batch inference job specification.

SaveMode(value[, names, module, qualname, ...])

Save mode options for batch inference output.