pointersequencer/experiment.py at main · landoskape/pointersequencer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
from matplotlib.pyplot import show
from ptrseq.experiments import create_experiment

if __name__ == "__main__":

    # Create experiment
    exp = create_experiment()

    if exp.args.showprms:
        # Load saved experiment (just experiment parameters)
        _ = exp.load_experiment(no_results=True)

        # Report parameters of saved experiment
        exp.report(args=True)

    elif not exp.args.justplot:
        # Report experiment details
        exp.report(init=True, args=True, meta_args=True)

        # Run main experiment method
        results, nets = exp.main()

        # Save results if requested
        if not exp.args.nosave:
            exp.save_experiment(results)

            # Save copy of repo
            exp.save_repo()

            # Save networks
            if exp.args.save_networks:
                exp.save_networks(nets)

    else:
        # Load saved experiment (parameters and results)
        print("Loading saved experiment...")
        results = exp.load_experiment(verbose=True)

        # Report saved experiment parameters
        exp.report(args=True)

    # Plot results unless calling script to show saved parameters
    if not exp.args.showprms:
        exp.plot(results)

        # Show all figures at end if requested
        if exp.args.showall:
            show()