Exemplo n.º 1
0
    logger.info("Running experiments")

    timings = {}

    if args.policy or args.all:
        print('\n\n')
        run_experiment(experiment_details, experiments.PolicyIterationExperiment, 'PI', verbose, timings, \
                       MAX_STEPS['pi'], NUM_TRIALS['pi'], theta=PI_THETA)

    if args.value or args.all:
        print('\n\n')
        run_experiment(experiment_details, experiments.ValueIterationExperiment, 'VI', verbose, timings, \
                       MAX_STEPS['vi'], NUM_TRIALS['vi'], theta=VI_THETA)

    if args.ql or args.all:
        print('\n\n')
        run_experiment(experiment_details, experiments.QLearnerExperiment, 'QL', verbose, timings, \
                       MAX_STEPS['ql'], NUM_TRIALS['ql'], max_episodes=QL_MAX_EPISODES, \
                       min_sub_thetas=QL_MIN_SUB_THETAS, theta=QL_THETA, epsilon_decays=QL_EPSILON_DECAYS) # NUM_TRIALS = 1000, MAX_STEPS = 30000, MAX_EPISIDES = 30000, QL_MIN_SUB_THETAS=5, QL_THETA = 0.001, QL_EPSILON_DECAYS = [0.00001]

    if args.plot:
        print('\n\n')
        if verbose:
            logger.info("----------")
        logger.info("Plotting results")
        plotting.plot_results(envs)

    print('\n\n')
    logger.info(timings)
    print('\n\n')
Exemplo n.º 2
0
            run_experiment(experiment_details, experiments.BenchmarkExperiment,
                           'Benchmark', args.dim, args.skiprerun, verbose,
                           timings)
        if args.ica or args.all:
            run_experiment(experiment_details, experiments.ICAExperiment,
                           'ICA', args.dim, args.skiprerun, verbose, timings)
        if args.pca or args.all:
            run_experiment(experiment_details, experiments.PCAExperiment,
                           'PCA', args.dim, args.skiprerun, verbose, timings)
        # if args.lda or args.all:
        #     run_experiment(experiment_details, experiments.LDAExperiment, 'LDA', args.dim, args.skiprerun,
        #                    verbose, timings)
        # if args.svd or args.all:
        #     run_experiment(experiment_details, experiments.SVDExperiment, 'SVD', args.dim, args.skiprerun,
        #                    verbose, timings)
        if args.rf or args.all:
            run_experiment(experiment_details, experiments.RFExperiment, 'RF',
                           args.dim, args.skiprerun, verbose, timings)
        if args.rp or args.all:
            run_experiment(experiment_details, experiments.RPExperiment, 'RP',
                           args.dim, args.skiprerun, verbose, timings)

        logger.info("Timings: {}".format(timings))

    if args.plot:
        if verbose:
            logger.info("----------")

        logger.info("Plotting results")
        plotting.plot_results()