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)