def feature_selection(data, features):
    """
    Choose which features to use for training.

    :param data: preprocessed dataset
    :param features: list of features to use
    :return: data with selected features
    """
    return data[features]