deepr.examples.multiply.jobs package

Submodules

deepr.examples.multiply.jobs.build module

Build a dummy dataset of random (x, 2*x) as a tfrecord file

class deepr.examples.multiply.jobs.build.Build(path_dataset, num_examples=1000)

Bases: Job

Build a dummy dataset of random (x, 2*x) as a tfrecord file

num_examples: int = 1000

path_dataset: str

run()

Run Job

deepr.examples.multiply.jobs.predict module

Compute prediction on a dataset and log result.

class deepr.examples.multiply.jobs.predict.PredictProto(path_model, graph_name, input_fn, prepro_fn, feeds, fetches)

Bases: Job

Compute predictions from a single .pb file.

feeds: Union[str, List[str]]

fetches: Union[str, List[str]]

graph_name: str

input_fn: Callable[DatasetV1]

path_model: str

prepro_fn: Callable[[DatasetV1, str], tensorflow.python.data.ops.dataset_ops.DatasetV1]

run()

Run Job

class deepr.examples.multiply.jobs.predict.PredictSavedModel(path_saved_model, input_fn, prepro_fn, feeds=None, fetches=None)

Bases: Job

Compute predictions from a directory containing saved models.

feeds: Optional[Union[str, List[str]]] = None

fetches: Optional[Union[str, List[str]]] = None

input_fn: Callable[DatasetV1]

path_saved_model: str

prepro_fn: Callable[[DatasetV1, str], tensorflow.python.data.ops.dataset_ops.DatasetV1]

run()

Run Job

Module contents