train

Documentation

Index

• Constants
• type Task
• type TrainManager
  • func NewTrainManager() *TrainManager
  • func (tm *TrainManager) CreateTask(ctx context.Context, dataPath, trainingDataPath string) (string, error)
  • func (tm *TrainManager) GetTaskStatus(ctx context.Context, taskID string) (string, error)

Constants

const (
	StatusCreated    = "Created"
	StatusProcessing = "Processing"
	StatusCompleted  = "Completed"
	StatusFailed     = "Failed" // Added for handling failure cases
)

Task statuses

Types

type Task

type Task struct {
	ID               string
	DataPath         string
	TrainingDataPath string
	Status           string
}

Task represents a task with an ID, data path, training data path, and status

type TrainManager

type TrainManager struct {
	// contains filtered or unexported fields
}

TrainManager manages tasks.

func NewTrainManager

func NewTrainManager() *TrainManager

NewTrainManager initializes a new TrainManager

func (*TrainManager) CreateTask

func (tm *TrainManager) CreateTask(ctx context.Context, dataPath, trainingDataPath string) (string, error)

CreateTask creates a new task and returns its ID

func (*TrainManager) GetTaskStatus

func (tm *TrainManager) GetTaskStatus(ctx context.Context, taskID string) (string, error)

GetTaskStatus retrieves the status of a task by its ID