data

Documentation

Index

• Constants
• type MLModel
• type Repository
  • func NewRepository(db postgresql.DB) *Repository
  • func (r *Repository) ChangeFeaturesCount(ctx context.Context, userID, modelType string, featuresCount int) error
  • func (r *Repository) CreateModel(ctx context.Context, userID, modelType string) error
  • func (r *Repository) GetModelByUserID(ctx context.Context, userID, modelType string) (*MLModel, error)
  • func (r *Repository) GetModelsByUserID(ctx context.Context, userID string) ([]MLModel, error)
  • func (r *Repository) SetFeaturesCountUsed(ctx context.Context, userID, modelType string, featuresCount int) error
  • func (r *Repository) SetModelS3Key(ctx context.Context, s3Key, modelType, userID string) error
  • func (r *Repository) SetModelStatus(ctx context.Context, status, modelType string, userID string) error
  • func (r *Repository) ViewNotFineTunedFaceModels(ctx context.Context, modelType string, tuneThreshold uint64) ([]MLModel, error)
  • func (r *Repository) ViewNotLearnedModels(ctx context.Context, modelType string, trainThreshold uint64) ([]MLModel, error)

Constants

const (
	StatusNotTrain       = "not_train"
	StatusInTrainProcess = "in_train_process"
	StatusInTuneProcess  = "in_tune_process"
	StatusTrained        = "train"
)

const (
	FaceModel = "face_model"
)

const (
	ModelsTable = "models"
)

Types

type MLModel

type MLModel struct {
	UserID           string  `db:"user_id"`
	ModelFeatures    uint64  `db:"features_count"`
	ModelTrainStatus string  `db:"train_status"`
	ModelType        string  `db:"model_type"`
	S3Key            *string `db:"s3_key"`
}

type Repository

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

func NewRepository

func NewRepository(db postgresql.DB) *Repository

func (*Repository) ChangeFeaturesCount

func (r *Repository) ChangeFeaturesCount(ctx context.Context, userID, modelType string, featuresCount int) error

func (*Repository) CreateModel

func (r *Repository) CreateModel(ctx context.Context, userID, modelType string) error

func (*Repository) GetModelByUserID

func (r *Repository) GetModelByUserID(ctx context.Context, userID, modelType string) (*MLModel, error)

func (*Repository) GetModelsByUserID

func (r *Repository) GetModelsByUserID(ctx context.Context, userID string) ([]MLModel, error)

func (*Repository) SetFeaturesCountUsed

func (r *Repository) SetFeaturesCountUsed(ctx context.Context, userID, modelType string, featuresCount int) error

func (*Repository) SetModelS3Key

func (r *Repository) SetModelS3Key(ctx context.Context, s3Key, modelType, userID string) error

func (*Repository) SetModelStatus

func (r *Repository) SetModelStatus(ctx context.Context, status, modelType string, userID string) error

func (*Repository) ViewNotFineTunedFaceModels

func (r *Repository) ViewNotFineTunedFaceModels(ctx context.Context, modelType string, tuneThreshold uint64) ([]MLModel, error)

func (*Repository) ViewNotLearnedModels

func (r *Repository) ViewNotLearnedModels(ctx context.Context, modelType string, trainThreshold uint64) ([]MLModel, error)