Skip to content

Commit 27665ce

Browse files
smurchingtomasatdatabricks
authored andcommitted
Update __init__.py to properly expose KerasTransformer via 'from sparkdl import KerasTransformer' (#96)
1 parent 94452d6 commit 27665ce

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

python/sparkdl/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
from .image.imageIO import imageSchema, imageType, readImages
1818
from .transformers.keras_image import KerasImageFileTransformer
1919
from .transformers.named_image import DeepImagePredictor, DeepImageFeaturizer
20+
from .transformers.keras_tensor import KerasTransformer
2021
from .transformers.tf_image import TFImageTransformer
2122
from .transformers.tf_tensor import TFTransformer
2223
from .transformers.utils import imageInputPlaceholder

0 commit comments

Comments
 (0)