snowflake.ml.model

Classes

Model()

Model Object containing multiple versions.

ModelVersion()

Model Version Object representing a specific version of the model that could be run.

HuggingFacePipelineModel([task, model, ...])

Utility factory method to build a wrapper over transformers [Pipeline].

snowflake.ml.model.custom_model

Classes

MethodRef(model_ref, method_name)

Represents a method invocation of an instance of ModelRef.

ModelRef(name, model)

Represents a model in the inference graph.

ModelContext(*[, artifacts, models])

Context for a custom model showing paths to artifacts and mapping between model name and object reference.

CustomModel([context])

Abstract class for user defined custom model.

snowflake.ml.model.model_signature

Classes

DataType(value)

An enumeration.

BaseFeatureSpec(name)

Abstract Class for specification of a feature.

FeatureSpec(name, dtype[, shape, nullable])

Specification of a feature in Snowflake native model packaging.

ModelSignature(inputs, outputs)

Signature of a model that specifies the input and output of a model.

Methods

infer_signature(input_data, output_data[, ...])

Infer model signature from given input and output sample data.