octomizer.models package

Submodules

octomizer.models.onnx_model module

A wrapper for ONNX Models in the OctoML Platform.

class octomizer.models.onnx_model.ONNXModel(client: octomizer.client.OctomizerClient, name: str, model: Optional[Union[bytes, str]] = None, description: Optional[str] = None, labels: Optional[List[str]] = None, timeout: int = 1200)

Bases: octomizer.model.Model

Represents a Model in ONNX format.

octomizer.models.tensorflow_graph_def_model module

A wrapper for TensorFlow Graph Def Models in the OctoML Platform.

class octomizer.models.tensorflow_graph_def_model.TensorFlowGraphDefModel(client: octomizer.client.OctomizerClient, name: str, model: Optional[Union[bytes, str]] = None, description: Optional[str] = None, labels: Optional[List[str]] = None, timeout: int = 1200)

Bases: octomizer.model.Model

Represents a Model in TensorFlow Graph Def format.

octomizer.models.tensorflow_keras_model module

A wrapper for TensorFlow Keras in the OctoML Platform.

class octomizer.models.tensorflow_keras_model.TensorFlowKerasModel(client: octomizer.client.OctomizerClient, name: str, model: Optional[Union[bytes, str]] = None, description: Optional[str] = None, labels: Optional[List[str]] = None, timeout: int = 1200)

Bases: octomizer.model.Model

Represents a Model in TensorFlow Keras format.

octomizer.models.tflite_model module

A wrapper for TFLite Models in the OctoML Platform.

class octomizer.models.tflite_model.TFLiteModel(client: octomizer.client.OctomizerClient, name: str, model: Optional[Union[bytes, str]] = None, description: Optional[str] = None, labels: Optional[List[str]] = None, timeout: int = 1200)

Bases: octomizer.model.Model

Represents a Model in TFLite format.

Module contents