trainer

Documentation

Index

• type ModelTrainer
  • func NewModelTrainer(trainRepo train_repository.Repository, connRepo conn_repository.Repository, ...) *ModelTrainer
  • func (mt *ModelTrainer) SaveResult() error
  • func (mt *ModelTrainer) Setup() (err error)

Types

type ModelTrainer

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

func NewModelTrainer

func NewModelTrainer(
	trainRepo train_repository.Repository,
	connRepo conn_repository.Repository,
	modelTrainingID string,
) *ModelTrainer

func (*ModelTrainer) SaveResult

func (mt *ModelTrainer) SaveResult() error

This function saves a training result. To do this, it performs the following steps:

1. It extracts the training entity from repository storage, for example, from the EDI server.
2. It creates the training zip archive.
2. It uploads the training zip archive to object storage.
4. Finally, it saves the training entity results.

func (*ModelTrainer) Setup

func (mt *ModelTrainer) Setup() (err error)

This function prepares a training environment. To do this, it performs the following steps:

1. It extracts the training entity from repository storage, for example, from the EDI server.
2. It downloads the source code of model training.
3. The setup function downloads all training data.
4. Finally, it saves the training entity to allow an ML toolchain to use it.