Versions in this module Expand all Collapse all v0 v0.26.28 Mar 22, 2022 Changes in this version + func GetTfkgPythonCode(customDefinitions []string) string + func GetVanillaPythonCode() string + type CompileConfig struct + BatchSize int + CpuInference bool + Loss Loss + ModelInfoSaveDir string + Optimizer optimizer.Optimizer + type Loss string + var LossBinaryCrossentropy Loss = "binary_crossentropy" + var LossMSE Loss = "mse" + var LossSparseCategoricalCrossentropy Loss = "sparse_categorical_crossentropy" + var Version = "0.2.6" + type TfkgModel struct + func LoadVanillaModel(errorHandler *cberrors.ErrorsContainer, logger *cblog.Logger, dir string, ...) (*TfkgModel, error) + func NewModel(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 v0.2.6 Dec 6, 2021 Changes in this version + type EvaluateConfig struct + BatchSize int + Callbacks []callback.Callback + Metrics []metric.Metric + PreFetch int + Verbose int + type FitConfig struct + BatchSize int + Callbacks []callback.Callback + Epochs int + Metrics []metric.Metric + PreFetch int + Validation bool + Verbose int + type TfModel struct + 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
Go to main page