models

Documentation

Index

• Constants
• Variables
• func AllModelVersionStages() string
• type ModelVersion
  • func (mv ModelVersion) ToProto() *protos.ModelVersion
• type ModelVersionStage
  • func (s ModelVersionStage) String() string
• type ModelVersionTag
• type RegisteredModel
• type RegisteredModelAlias
• type RegisteredModelTag

Constants

const (
	ModelVersionStageNone       = "None"
	ModelVersionStageStaging    = "Staging"
	ModelVersionStageProduction = "Production"
	ModelVersionStageArchived   = "Archived"
)

const StageDeletedInternal = "Deleted_Internal"

Variables

var CanonicalMapping = map[string]string{
	strings.ToLower(ModelVersionStageNone):       ModelVersionStageNone,
	strings.ToLower(ModelVersionStageStaging):    ModelVersionStageStaging,
	strings.ToLower(ModelVersionStageProduction): ModelVersionStageProduction,
	strings.ToLower(ModelVersionStageArchived):   ModelVersionStageArchived,
}

Functions

func AllModelVersionStages

func AllModelVersionStages() string

Types

type ModelVersion

type ModelVersion struct {
	Name            string            `db:"name"              gorm:"column:name;primaryKey"`
	Version         int32             `db:"version"           gorm:"column:version;primaryKey"`
	CreationTime    int64             `db:"creation_time"     gorm:"column:creation_time"`
	LastUpdatedTime int64             `db:"last_updated_time" gorm:"column:last_updated_time"`
	Description     string            `db:"description"       gorm:"column:description"`
	UserID          string            `db:"user_id"           gorm:"column:user_id"`
	CurrentStage    ModelVersionStage `db:"current_stage"     gorm:"column:current_stage"`
	Source          string            `db:"source"            gorm:"column:source"`
	RunID           string            `db:"run_id"            gorm:"column:run_id"`
	Status          string            `db:"status"            gorm:"column:status"`
	StatusMessage   string            `db:"status_message"    gorm:"column:status_message"`
	RunLink         string            `db:"run_link"          gorm:"column:run_link"`
	StorageLocation string            `db:"storage_location"  gorm:"column:storage_location"`
}

ModelVersion mapped from table <model_versions>.

func (ModelVersion) ToProto

func (mv ModelVersion) ToProto() *protos.ModelVersion

type ModelVersionStage

type ModelVersionStage string

func (ModelVersionStage) String

func (s ModelVersionStage) String() string

type ModelVersionTag

type ModelVersionTag struct {
	Key     string `db:"key"     gorm:"column:key;primaryKey"`
	Value   string `db:"value"   gorm:"column:value"`
	Name    string `db:"name"    gorm:"column:name;primaryKey"`
	Version int32  `db:"version" gorm:"column:version;primaryKey"`
}

ModelVersionTag mapped from table <model_version_tags>.

type RegisteredModel

type RegisteredModel struct {
	Name            string `db:"name"              gorm:"column:name;primaryKey"`
	CreationTime    int64  `db:"creation_time"     gorm:"column:creation_time"`
	LastUpdatedTime int64  `db:"last_updated_time" gorm:"column:last_updated_time"`
	Description     string `db:"description"       gorm:"column:description"`
}

RegisteredModel mapped from table <registered_models>.

type RegisteredModelAlias

type RegisteredModelAlias struct {
	Alias   string `db:"alias"   gorm:"column:alias;primaryKey"`
	Version int32  `db:"version" gorm:"column:version;not null"`
	Name    string `db:"name"    gorm:"column:name;primaryKey"`
}

RegisteredModelAlias mapped from table <registered_model_aliases>.

type RegisteredModelTag

type RegisteredModelTag struct {
	Key   string `db:"key"   gorm:"column:key;primaryKey"`
	Value string `db:"value" gorm:"column:value"`
	Name  string `db:"name"  gorm:"column:name;primaryKey"`
}

RegisteredModelTag mapped from table <registered_model_tags>.