def test_structured_data_block():
block = hyperblock_module.StructuredDataBlock()
block.num_heads = 1
block.set_state(block.get_state())
hp = kerastuner.HyperParameters()
block.build(hp, ak.Input())
assert common.name_in_hps('block_type', hp)
def assemble(self, input_node):
self.infer_column_types()
if input_node.column_types is None:
input_node.column_types = self.column_types
# partial column_types is provided.
for key, value in self.column_types.items():
if key not in input_node.column_types:
input_node.column_types[key] = value
return hyperblock.StructuredDataBlock(seed=self.seed)(input_node)
def test_structured_data_block():
block = hyperblock_module.StructuredDataBlock()
block.heads = [ak.ClassificationHead()]
block.set_state(block.get_state())
hp = kerastuner.HyperParameters()
block.build(hp, ak.Input())
assert common.name_in_hps('module_type', hp)
def assemble(self, input_node):
return hyperblock.StructuredDataBlock(seed=self.seed)(input_node)
