model

Documentation

Index

• type EvaluateConfig
• type FitConfig
• type TfModel
  • func LoadModel(errorHandler *cberrors.ErrorsContainer, logger *cblog.Logger, dir string) (*TfModel, error)
  • func NewSequentialModel(logger *cblog.Logger, errorHandler *cberrors.ErrorsContainer, ...) *TfModel
  • func (m *TfModel) CompileAndLoad(batchSize int, pythonPath string, ...) error
  • func (m *TfModel) Evaluate(evaluateSignature string, mode data.GeneratorMode, dataset data.Dataset, ...)
  • func (m *TfModel) Fit(trainSignature string, valSignature string, dataset data.Dataset, ...)
  • func (m *TfModel) Save(dir string) error

Types

type EvaluateConfig

type EvaluateConfig struct {
	BatchSize int
	PreFetch  int
	Metrics   []metric.Metric
	Callbacks []callback.Callback
	Verbose   int
}

type FitConfig

type FitConfig struct {
	Epochs     int
	BatchSize  int
	Validation bool
	PreFetch   int
	Metrics    []metric.Metric
	Callbacks  []callback.Callback
	Verbose    int
}

type TfModel

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

func LoadModel

func LoadModel(
	errorHandler *cberrors.ErrorsContainer,
	logger *cblog.Logger,
	dir string,
) (*TfModel, error)

func NewSequentialModel

func NewSequentialModel(
	logger *cblog.Logger,
	errorHandler *cberrors.ErrorsContainer,
	layers ...layer.Layer,
) *TfModel

func (*TfModel) CompileAndLoad

func (m *TfModel) CompileAndLoad(batchSize int, pythonPath string, createModelFunc ...func(tempDir, tempPythonGenerationFile string) error) error

func (*TfModel) Evaluate

func (m *TfModel) Evaluate(
	evaluateSignature string,
	mode data.GeneratorMode,
	dataset data.Dataset,
	config EvaluateConfig,
)

func (*TfModel) Fit

func (m *TfModel) Fit(
	trainSignature string,
	valSignature string,
	dataset data.Dataset,
	config FitConfig,
)

func (*TfModel) Save

func (m *TfModel) Save(dir string) error