sourced.ml.core.models.tensorflow

Module Contents

class sourced.ml.core.models.tensorflow.TensorFlowModel[source]

Bases:modelforge.Model

TensorFlow Protobuf model exported in the Modelforge format with GraphDef inside.

NAME = tensorflow-model[source]
VENDOR = source{d}[source]
DESCRIPTION = TensorFlow Protobuf model that contains a GraphDef instance.[source]
LICENSE[source]
graphdef[source]

Returns the wrapped TensorFlow GraphDef.

construct(self, graphdef: 'tensorflow.GraphDef' = None, session: 'tensorflow.Session' = None, outputs: List[str] = None)[source]