def get_params()

in run_experiments_real.py [0:0]

• 11 lines of code
• 4 McCabe index (conditional complexity)

def get_params(args):
    params = BASE_PARAMS
    if args.data == "mnist":
        params["batch_size"] = [8, 16, 32, 64]
    elif args.data == "shapes":
        params["batch_size"] = [4, 8, 16, 32]

    if args.model == "cci_vae":
        params["n_epochs"] = [10, 20, 50]
        params["beta"] = [4.0, 10.0, 100.0, 1000.0]
        params["z_dim"] = [10, 30]
    return params