train

package
v0.0.0-...-b0e91a5 Latest Latest
Warning

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

Go to latest
Published: Oct 6, 2022 License: Apache-2.0 Imports: 15 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func List

func List(bucketName string) error

func ListArchives

func ListArchives(ctx context.Context, bucket string) error

func ListJob

func ListJob(ctx context.Context) error

Types

type ModelType

type ModelType int
const (
	ModelTypeUnknown ModelType = iota
	ModelTypeCategorical
	ModelTypeLinear
)

func ParseModelType

func ParseModelType(s string) ModelType

func (ModelType) String

func (m ModelType) String() string

type Training

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

func New

func New(bucketName string, ociImage, roleArn string) *Training

func (*Training) GetTrainingOutput

func (t *Training) GetTrainingOutput(ctx context.Context, jobName, outputFile string) error

func (*Training) TrainDir

func (t *Training) TrainDir(ctx context.Context, jobName, basedir string, modelType ModelType, imgWidth, imgHeight, sliceSize int, horizon int, withFlipImage bool, outputModelFile string, enableSpotTraining bool) error

func (Training) UploadArchive

func (t Training) UploadArchive(ctx context.Context, archive []byte) error

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL