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 storing paths to file artifacts and model object references.

CustomModel([context])

Abstract class for user defined custom model.

snowflake.ml.model.model_signature

Classes

DataType(value)

An enumeration.

BaseFeatureSpec(name, shape)

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.

snowflake.ml.model.openai_signatures

Attributes

OPENAI_CHAT_SIGNATURE

dict() -> new empty dictionary dict(mapping) -> new dictionary initialized from a mapping object's (key, value) pairs dict(iterable) -> new dictionary initialized as if via: d = {} for k, v in iterable: d[k] = v dict(**kwargs) -> new dictionary initialized with the name=value pairs in the keyword argument list. For example: dict(one=1, two=2).