Ejemplos de StructuredDataAssembler en Python

Ejemplo n.º 1

Archivo: test_meta_model.py Proyecto: wade1990/autokeras

def test_structured_data_assembler():
    data = common.structured_data()
    dataset = tf.data.Dataset.from_tensor_slices(data)
    assembler = meta_model.StructuredDataAssembler(
        column_names=common.COLUMN_NAMES_FROM_NUMPY)
    for line in dataset:
        assembler.update(line)

    assembler.assemble(node.StructuredDataInput())

Ejemplo n.º 2

Archivo: meta_model_test.py Proyecto: zhaoqiuye/autokeras

def test_structured_data_assembler():
    data = common.generate_structured_data()
    dataset = tf.data.Dataset.from_tensor_slices(data)
    assembler = meta_model.StructuredDataAssembler()
    for line in dataset:
        assembler.update(line)

    input_node = node.StructuredDataInput()
    assembler.assemble(input_node)
    assert isinstance(input_node.out_blocks[0], ak.StructuredDataBlock)