Skip to content

Commit

Permalink
Preprocessing docstring and cli_test update
Browse files Browse the repository at this point in the history
  • Loading branch information
jamescook106 committed Aug 8, 2019
1 parent 05c1976 commit ca2b082
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 1 deletion.
3 changes: 2 additions & 1 deletion zookeeper/cli_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,8 @@ def outputs(self, data):

@registry.register_preprocess("mnist")
class default(ImageClassification):
pass
def inputs(self, data):
return super().inputs(data) / 255


@registry.register_preprocess("mnist")
Expand Down
11 changes: 11 additions & 0 deletions zookeeper/preprocessing.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,17 @@


class Preprocessing(abc.ABC):
""" An abstract Preprocessing class that contains the abstract contains the
methods required for Preprocessing.
We define a neural network to be a mapping between an input and an output.
hence we define to abstract methods, an input (e.g the image for image
classification), and an output (e.g the class label for image classification).
We also define decoders, which allows use to customize the decoding, and kwargs,
which allows one to pass information from the preprocessing to the model (e.g
input size.)
"""

decoders = None
kwargs = {}

Expand Down

0 comments on commit ca2b082

Please sign in to comment.