model

Documentation

Index

• Constants
• type MlflowClient
  • func NewMlflowClient(trackingUri, username, password string) *MlflowClient
  • func NewProxiedMlflowClient(configr *config.ArenaConfiger, service *corev1.Service, username string, ...) *MlflowClient
  • func (c *MlflowClient) CheckHealth() (bool, error)
  • func (c *MlflowClient) CreateModelVersion(name, source, runId string, tags []*types.ModelVersionTag, ...) (*types.ModelVersion, error)
  • func (c *MlflowClient) CreateRegisteredModel(name string, tags []*types.RegisteredModelTag, description string) (*types.RegisteredModel, error)
  • func (c *MlflowClient) CreateRegisteredModelAndModelVersion(name string, description string, tags []*types.RegisteredModelTag, ...) (*types.RegisteredModel, *types.ModelVersion, error)
  • func (c *MlflowClient) DeleteModelVersion(name string, version string) error
  • func (c *MlflowClient) DeleteModelVersionTag(name, version, key string) error
  • func (c *MlflowClient) DeleteRegisteredModel(name string) error
  • func (c *MlflowClient) DeleteRegisteredModelAlias(name, alias string) error
  • func (c *MlflowClient) DeleteRegisteredModelTag(name, key string) error
  • func (c *MlflowClient) GetDownloadUri(name, version string) (string, error)
  • func (c *MlflowClient) GetLatestModelVersions(name string, stages []string) ([]*types.ModelVersion, error)
  • func (c *MlflowClient) GetModelVersion(name, version string) (*types.ModelVersion, error)
  • func (c *MlflowClient) GetModelVersionByAlias(name, alias string) (*types.ModelVersion, error)
  • func (c *MlflowClient) GetRegisteredModel(name string) (*types.RegisteredModel, error)
  • func (c *MlflowClient) RenameRegisteredModel(name, newName string) (*types.RegisteredModel, error)
  • func (c *MlflowClient) SearchModelVersions(filter string, maxResults int, orderBy []string) ([]*types.ModelVersion, error)
  • func (c *MlflowClient) SearchRegisteredModels(filter string, maxResults int, orderBy []string) ([]*types.RegisteredModel, error)
  • func (c *MlflowClient) SetModelVersionTag(name, version, key, value string) error
  • func (c *MlflowClient) SetRegisteredModelAlias(name, version, alias string) error
  • func (c *MlflowClient) SetRegisteredModelTag(name, key, value string) error
  • func (c *MlflowClient) UpdateModelVersion(name string, version string, description string) (*types.ModelVersion, error)
  • func (c *MlflowClient) UpdateRegisteredModel(name string, description string) (*types.RegisteredModel, error)
• type MlflowError
  • func (e *MlflowError) Error() string
  • func (e *MlflowError) IsResourceAlreadyExistsError() bool
  • func (e *MlflowError) IsResourceDoesNotExistError() bool

Constants

const (
	RESOURCE_DOES_NOT_EXIST_ERROR = "RESOURCE_DOES_NOT_EXIST"
	RESOURCE_ALREADY_EXISTS_ERROR = "RESOURCE_ALREADY_EXISTS"
)

Types

type MlflowClient

type MlflowClient struct {
	RestyClient *resty.Client
}

func NewMlflowClient

func NewMlflowClient(trackingUri, username, password string) *MlflowClient

func NewProxiedMlflowClient

func NewProxiedMlflowClient(configr *config.ArenaConfiger, service *corev1.Service, username string, password string) *MlflowClient

Create a MLflow client proxied by Kubernetes api server

func (*MlflowClient) CheckHealth

func (c *MlflowClient) CheckHealth() (bool, error)

func (*MlflowClient) CreateModelVersion

func (c *MlflowClient) CreateModelVersion(name, source, runId string, tags []*types.ModelVersionTag, runLink, description string) (*types.ModelVersion, error)

func (*MlflowClient) CreateRegisteredModel

func (c *MlflowClient) CreateRegisteredModel(name string, tags []*types.RegisteredModelTag, description string) (*types.RegisteredModel, error)

For detailed information about MLflow REST API, see https://mlflow.org/docs/latest/rest-api.html

func (*MlflowClient) CreateRegisteredModelAndModelVersion

func (c *MlflowClient) CreateRegisteredModelAndModelVersion(
	name string,
	description string,
	tags []*types.RegisteredModelTag,
	version string,
	versionDescription string,
	versionTags []*types.ModelVersionTag,
	source string,
) (*types.RegisteredModel, *types.ModelVersion, error)

func (*MlflowClient) DeleteModelVersion

func (c *MlflowClient) DeleteModelVersion(name string, version string) error

func (*MlflowClient) DeleteModelVersionTag

func (c *MlflowClient) DeleteModelVersionTag(name, version, key string) error

func (*MlflowClient) DeleteRegisteredModel

func (c *MlflowClient) DeleteRegisteredModel(name string) error

func (*MlflowClient) DeleteRegisteredModelAlias

func (c *MlflowClient) DeleteRegisteredModelAlias(name, alias string) error

func (*MlflowClient) DeleteRegisteredModelTag

func (c *MlflowClient) DeleteRegisteredModelTag(name, key string) error

func (*MlflowClient) GetDownloadUri

func (c *MlflowClient) GetDownloadUri(name, version string) (string, error)

func (*MlflowClient) GetLatestModelVersions

func (c *MlflowClient) GetLatestModelVersions(name string, stages []string) ([]*types.ModelVersion, error)

func (*MlflowClient) GetModelVersion

func (c *MlflowClient) GetModelVersion(name, version string) (*types.ModelVersion, error)

func (*MlflowClient) GetModelVersionByAlias

func (c *MlflowClient) GetModelVersionByAlias(name, alias string) (*types.ModelVersion, error)

func (*MlflowClient) GetRegisteredModel

func (c *MlflowClient) GetRegisteredModel(name string) (*types.RegisteredModel, error)

func (*MlflowClient) RenameRegisteredModel

func (c *MlflowClient) RenameRegisteredModel(name, newName string) (*types.RegisteredModel, error)

func (*MlflowClient) SearchModelVersions

func (c *MlflowClient) SearchModelVersions(filter string, maxResults int, orderBy []string) ([]*types.ModelVersion, error)

func (*MlflowClient) SearchRegisteredModels

func (c *MlflowClient) SearchRegisteredModels(filter string, maxResults int, orderBy []string) ([]*types.RegisteredModel, error)

func (*MlflowClient) SetModelVersionTag

func (c *MlflowClient) SetModelVersionTag(name, version, key, value string) error

func (*MlflowClient) SetRegisteredModelAlias

func (c *MlflowClient) SetRegisteredModelAlias(name, version, alias string) error

func (*MlflowClient) SetRegisteredModelTag

func (c *MlflowClient) SetRegisteredModelTag(name, key, value string) error

func (*MlflowClient) UpdateModelVersion

func (c *MlflowClient) UpdateModelVersion(name string, version string, description string) (*types.ModelVersion, error)

func (*MlflowClient) UpdateRegisteredModel

func (c *MlflowClient) UpdateRegisteredModel(name string, description string) (*types.RegisteredModel, error)

type MlflowError

type MlflowError struct {
	ErrorCode string `json:"error_code"`
	Message   string `json:"message"`
}

func (*MlflowError) Error

func (e *MlflowError) Error() string

func (*MlflowError) IsResourceAlreadyExistsError

func (e *MlflowError) IsResourceAlreadyExistsError() bool

func (*MlflowError) IsResourceDoesNotExistError

func (e *MlflowError) IsResourceDoesNotExistError() bool