def feature_selection(all_features):
    """
    Description: Function defining which features to use
    Input: all_features -> boolean telling us if we use all the features or only the important ones
    Return: A list of the features we keep
    """
    if all_features:
        features = ['Age', 'HoursPerWeek', 'TotalHours', 'APM', 'SelectByHotkeys', 'AssignToHotkeys','UniqueHotkeys',
            'MinimapAttacks', 'MinimapRightClicks', 'NumberOfPACs','GapBetweenPACs', 'ActionLatency', 'ActionsInPAC', 
            'TotalMapExplored', 'WorkersMade', 'UniqueUnitsMade', 'ComplexUnitsMade','ComplexAbilitiesUsed']
    else:
        features = ['APM', 'SelectByHotkeys', 'AssignToHotkeys', 'ActionLatency', 'GapBetweenPACs']
    
    return features