ensemble.py 文件源码

def get_ensemble_weights(ensemble_dirs):
    '''
    return ensembling weightes by reading ./output/<ensemble_dir>/models_ensembled.txt
    '''
    total_models = 0
    weights = np.zeros(len(ensemble_dirs))

    for i, ensemble_dir in enumerate(ensemble_dirs):
        ensembled_model_names, _ = get_models_ensembled(ensemble_dir)
        num_models_used = len(ensembled_model_names)

        total_models += num_models_used
        weights[i] = num_models_used

    weights = np.divide(weights, total_models)
    return weights