model

package
v0.9.15 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Apr 28, 2024 License: Apache-2.0 Imports: 13 Imported by: 1

Documentation

Index

Constants

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

Variables

This section is empty.

Functions

This section is empty.

Types

type MlflowClient added in v0.9.14 

type MlflowClient struct {
	RestyClient *resty.Client
}

func NewMlflowClient added in v0.9.14 

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

func NewProxiedMlflowClient added in v0.9.14 

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 added in v0.9.14 

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

func (*MlflowClient) CreateModelVersion added in v0.9.14 

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

func (*MlflowClient) CreateRegisteredModel added in v0.9.14 

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 added in v0.9.14 

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 added in v0.9.14 

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

func (*MlflowClient) DeleteModelVersionTag added in v0.9.14 

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

func (*MlflowClient) DeleteRegisteredModel added in v0.9.14 

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

func (*MlflowClient) DeleteRegisteredModelAlias added in v0.9.14 

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

func (*MlflowClient) DeleteRegisteredModelTag added in v0.9.14 

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

func (*MlflowClient) GetDownloadUri added in v0.9.14 

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

func (*MlflowClient) GetLatestModelVersions added in v0.9.14 

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

func (*MlflowClient) GetModelVersion added in v0.9.14 

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

func (*MlflowClient) GetModelVersionByAlias added in v0.9.14 

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

func (*MlflowClient) GetRegisteredModel added in v0.9.14 

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

func (*MlflowClient) RenameRegisteredModel added in v0.9.14 

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

func (*MlflowClient) SearchModelVersions added in v0.9.14 

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

func (*MlflowClient) SearchRegisteredModels added in v0.9.14 

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

func (*MlflowClient) SetModelVersionTag added in v0.9.14 

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

func (*MlflowClient) SetRegisteredModelAlias added in v0.9.14 

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

func (*MlflowClient) SetRegisteredModelTag added in v0.9.14 

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

func (*MlflowClient) UpdateModelVersion added in v0.9.14 

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

func (*MlflowClient) UpdateRegisteredModel added in v0.9.14 

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

type MlflowError added in v0.9.14 

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

func (*MlflowError) Error added in v0.9.14 

func (e *MlflowError) Error() string

func (*MlflowError) IsResourceAlreadyExistsError added in v0.9.14 

func (e *MlflowError) IsResourceAlreadyExistsError() bool

func (*MlflowError) IsResourceDoesNotExistError added in v0.9.14 

func (e *MlflowError) IsResourceDoesNotExistError() bool

Source Files

View all Source files

Directories

Path Synopsis

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.