d554d2ce78425cc70f690ad76972949c94fc6a7e,tpot/tpot.py,,main,#,738

Before Change

    tpot.fit(training_features, training_classes)

    if args.VERBOSITY >= 1:
        print("\nTraining accuracy: {}".format(tpot.score(training_features, training_classes)))
        print("Holdout accuracy: {}".format(tpot.score(testing_features, testing_classes)))

    if args.OUTPUT_FILE != "":

After Change

    tpot.fit(training_features, training_classes)

    if args.VERBOSITY in [1, 2] and tpot._optimized_pipeline:
        print("\nTraining score: {}".format(max([tpot.hof.keys[x].wvalues[1] for x in range(len(tpot.hof.keys))])))
        print("Holdout score: {}".format(tpot.score(testing_features, testing_classes)))
    elif args.VERBOSITY >= 3 and tpot.hof:
        print("\nFinal Pareto front testing scores:")
        for pipeline_num, pipeline in enumerate(tpot.hof.items):
            print("{}\t{}\t{}".format(-tpot.hof.keys[pipeline_num].wvalues[0],
                                      tpot.hof_fitted_pipelines[str(pipeline)].score(testing_features, testing_classes),
                                      pipeline))

    if args.OUTPUT_FILE != "":
        tpot.export(args.OUTPUT_FILE)

Instances