optimise.py 文件源码

def skoptTUNE(args, model, n_calls):
    """
    Hyper-parameter optimization using scikit-opt.
    It has 3 algorithms: forest_minimize (decision-tree regression search);
    gbrt_minimize (gradient-boosted-tree search);
    and hp_minimize (Gaussian process regression search).
    """
    hyperparameters = {
        'batch_size': (40, 120),
        'num_hidden': (100, 500),
        'dropout_output': (0.3, 1.0),
        'dropout_input': (0.3, 1.0),
        'clip_norm': (0.5, 1.0),
    }

    data = load_data(args, args.data, saved=args.load_data)
    all_res = skopt_search(args, data, model, hyperparameters, gp_minimize, n_calls=n_calls)
    print(all_res)