d6c1b86594ef9a97e9f503547ab6567f89508486,tests/integration_tests/test_experiment.py,,test_experiment_attention,#Any#,253

Before Change

    // Machine translation with attention
    input_features = "[{name: english, type: sequence, vocab_size: 10," \
                     " max_len: 10, encoder: rnn, cell_type: lstm} ]"
    output_features = Template("[{name: spanish, type: sequence,"
                               " vocab_size: 10, max_len: 10,"
                               " decoder: generator, cell_type: lstm,"
                               " attention: ${attention}}] ")

    // Generate test data
    rel_path = generate_data(
        input_features,
        output_features.substitute(attention="bahdanau"),
        csv_filename
    )

    for attention in ["bahdanau", "luong"]:
        run_experiment(input_features, output_features.substitute(
            attention=attention), data_csv=rel_path)


def test_experiment_sequence_combiner(csv_filename):

After Change

    input_features = [
            sequence_feature(encoder="rnn", cell_type="lstm", max_len=10)
        ]
    output_features = [
        sequence_feature(
            max_len=10,
            cell_type="lstm",
            decoder="generator",
            attention="bahdanau"
        )
    ]

    // Generate test data
    rel_path = generate_data(input_features, output_features, csv_filename)

    for attention in ["bahdanau", "luong"]:
        output_features[0]["attention"] = attention
        run_experiment(input_features, output_features, data_csv=rel_path)


def test_experiment_sequence_combiner(csv_filename):

Instances