def prune_features()

in src/run_paraphrase.py [0:0]


def prune_features(feat_dicts):
    return [{k: v for k, v in d.items() if int(k.split('-')[1]) in FEATURE_LAYERS}
            for d in feat_dicts]