You are viewing documentation about an older version (1.7.0). View latest version

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])	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.