d6c1b86594ef9a97e9f503547ab6567f89508486,tests/integration_tests/test_experiment.py,,test_experiment_model_resume,#Any#,316

Before Change

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

    model_definition = model_definition_template.substitute(
        input_name=input_features, output_name=output_features
    )

    exp_dir_name = experiment(yaml.safe_load(model_definition),
                              data_csv=rel_path)
    logging.info("Experiment Directory: {0}".format(exp_dir_name))

After Change

def test_experiment_model_resume(csv_filename):
    // Single sequence input, single category output
    // Tests saving a model file, loading it to rerun training and predict
    input_features = [sequence_feature(encoder="rnn", reduce_output="sum")]
    output_features = [categorical_feature(vocab_size=2, reduce_input="sum")]
    // Generate test data
    rel_path = generate_data(input_features, output_features, csv_filename)

    model_definition = {
        "input_features": input_features,
        "output_features": output_features,
        "combiner": {"type": "concat", "fc_size": 14},
        "training": {"epochs": 2}
    }

    exp_dir_name = experiment(model_definition, data_csv=rel_path)
    logging.info("Experiment Directory: {0}".format(exp_dir_name))

Instances