tensorflow

package

v0.0.0-...-8a61e0c Latest Latest Go to latest Published: Dec 21, 2020 License: Apache-2.0 Imports: 11 Imported by: 0

Details

This package provides an abstraction to tensorflow training which is carried out in python rather than go.

View Source

const CONFIG_FILE string = "config.in"

View Source

const OUTPUT_FILE string = "output.out"

View Source

const WEIGHTS_FILE string = "weights.in"

This section is empty.

func EnableDebug(b bool)

func NewTensorflowMLF() training.MLFramework

type TensorflowError struct {
	Err          error
	Description  string
	PythonOutput string
}

func (e *TensorflowError) Error() string

type TensorflowFramework struct {
}

func (t *TensorflowFramework) Aggregate(updates []string) (aggregatedWeights string, err error)

func (t *TensorflowFramework) Evaluate(configuration string, weights string) (results training.EvaluationResults, err error)

func (t *TensorflowFramework) Train(configuration string, weights string) (updatedWeights string, err error)

Trains the supplied model with local data and returns the updated weights