model

Documentation

Index

• func GetTfkgPythonCode(customDefinitions []string) string
• func GetVanillaPythonCode() string
• type CompileConfig
• type EvaluateConfig
• type FitConfig
• type Loss
• type TfkgModel
  • func LoadModel(errorHandler *cberrors.ErrorsContainer, logger *cblog.Logger, dir string, ...) (*TfkgModel, error)
  • func LoadVanillaModel(errorHandler *cberrors.ErrorsContainer, logger *cblog.Logger, dir string, ...) (*TfkgModel, error)
  • func NewModel(logger *cblog.Logger, errorHandler *cberrors.ErrorsContainer, ...) *TfkgModel
  • func NewSequentialModel(logger *cblog.Logger, errorHandler *cberrors.ErrorsContainer, ...) *TfkgModel
  • func (m *TfkgModel) CompileAndLoad(config CompileConfig, sessionOptions ...*for_core_protos_go_proto.ConfigProto) error
  • func (m *TfkgModel) Evaluate(mode data.GeneratorMode, dataset data.Dataset, config EvaluateConfig)
  • func (m *TfkgModel) Fit(dataset data.Dataset, config FitConfig)
  • func (m *TfkgModel) GetLayerWeights(layerName string) ([]*tf.Tensor, error)
  • func (m *TfkgModel) GetModelWeights() ([]*tf.Tensor, error)
  • func (m *TfkgModel) Predict(inputs ...*tf.Tensor) (*tf.Tensor, error)
  • func (m *TfkgModel) Save(dir string) error
  • func (m *TfkgModel) SetModelWeights(weights []*tf.Tensor) error

Functions

func GetTfkgPythonCode

func GetTfkgPythonCode(customDefinitions []string) string

This code is generated automatically using "go generate ./..." from model/tfkg_model.py. DO NOT EDIT manually.

func GetVanillaPythonCode

func GetVanillaPythonCode() string

Types

type CompileConfig

type CompileConfig struct {
	Loss             Loss
	Optimizer        optimizer.Optimizer
	ModelInfoSaveDir string
	BatchSize        int
	CpuInference     bool
}

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 Loss

type Loss string

var (
	Version = "0.2.6"

	LossBinaryCrossentropy            Loss = "binary_crossentropy"
	LossSparseCategoricalCrossentropy Loss = "sparse_categorical_crossentropy"
	LossMSE                           Loss = "mse"
)

type TfkgModel

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

func LoadModel

func LoadModel(
	errorHandler *cberrors.ErrorsContainer,
	logger *cblog.Logger,
	dir string,
	sessionOptions ...*for_core_protos_go_proto.ConfigProto,
) (*TfkgModel, error)

func LoadVanillaModel

func LoadVanillaModel(
	errorHandler *cberrors.ErrorsContainer,
	logger *cblog.Logger,
	dir string,
	loss Loss,
	optimizer optimizer.Optimizer,
	sessionOptions ...*for_core_protos_go_proto.ConfigProto,
) (*TfkgModel, error)

func NewModel

func NewModel(
	logger *cblog.Logger,
	errorHandler *cberrors.ErrorsContainer,
	output layer.Layer,
) *TfkgModel

func NewSequentialModel

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

func (*TfkgModel) CompileAndLoad

func (m *TfkgModel) CompileAndLoad(config CompileConfig, sessionOptions ...*for_core_protos_go_proto.ConfigProto) error

func (*TfkgModel) Evaluate

func (m *TfkgModel) Evaluate(
	mode data.GeneratorMode,
	dataset data.Dataset,
	config EvaluateConfig,
)

func (*TfkgModel) Fit

func (m *TfkgModel) Fit(
	dataset data.Dataset,
	config FitConfig,
)

func (*TfkgModel) GetLayerWeights

func (m *TfkgModel) GetLayerWeights(layerName string) ([]*tf.Tensor, error)

func (*TfkgModel) GetModelWeights

func (m *TfkgModel) GetModelWeights() ([]*tf.Tensor, error)

func (*TfkgModel) Predict

func (m *TfkgModel) Predict(inputs ...*tf.Tensor) (*tf.Tensor, error)

func (*TfkgModel) Save

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

func (*TfkgModel) SetModelWeights

func (m *TfkgModel) SetModelWeights(weights []*tf.Tensor) error