mlflow

Documentation

Index

• type Config
• type DeleteExperimentErrorResponse
• type DeleteExperimentRequest
• type DeleteRunRequest
• type RunData
• type RunInfo
• type RunResponse
• type RunTag
• type SearchRunResponse
• type SearchRunsRequest
• type SearchRunsResponse
• type Service
  • func NewMlflowService(httpClient *http.Client, config Config) (Service, error)

Types

type Config

type Config struct {
	TrackingURL         string
	ArtifactServiceType string
}

type DeleteExperimentErrorResponse

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

type DeleteExperimentRequest

type DeleteExperimentRequest struct {
	ExperimentID string `json:"experiment_id" required:"true"`
}

type DeleteRunRequest

type DeleteRunRequest struct {
	RunID string `json:"run_id" required:"true"`
}

type RunData

type RunData struct {
	Tags []RunTag `json:"tags"`
}

type RunInfo

type RunInfo struct {
	RunID          string `json:"run_id"`
	ExperimentID   string `json:"experiment_id"`
	UserID         string `json:"user_id"`
	LifecycleStage string `json:"lifecycle_stage"`
	ArtifactURI    string `json:"artifact_uri"`
}

type RunResponse

type RunResponse struct {
	Info RunInfo `json:"info"`
	Data RunData `json:"data"`
}

type RunTag

type RunTag struct {
	Key   string `json:"key"`
	Value string `json:"value"`
}

type SearchRunResponse

type SearchRunResponse struct {
	RunData RunResponse `json:"run"`
}

type SearchRunsRequest

type SearchRunsRequest struct {
	ExperimentID []string `json:"experiment_ids" required:"true"`
}

type SearchRunsResponse

type SearchRunsResponse struct {
	RunsData []RunResponse `json:"runs"`
}

type Service

type Service interface {
	DeleteExperiment(ctx context.Context, ExperimentID string, deleteArtifact bool) error
	DeleteRun(ctx context.Context, RunID, artifactURL string, deleteArtifact bool) error
}

func NewMlflowService

func NewMlflowService(httpClient *http.Client, config Config) (Service, error)