bug fix

keras-team · Sep 24, 2019 · e07093a · e07093a
1 parent 6a47d1c
commit e07093a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/autokeras/hypermodel/preprocessor.py b/autokeras/hypermodel/preprocessor.py
@@ -539,6 +539,9 @@ def output_types(self):
     def output_shape(self):
         return self.lightgbm_block.output_shape
 
+    def set_hp(self, hp):
+        self.lightgbm_block.set_hp(hp)
+
 
 class ImageAugmentation(Preprocessor):
     """Collection of various image augmentation methods.