# This class allows us to create a set of input features
# where each new input feature is added as another dimension
# For example
# We could add a value for the first element related to age
# and in the second element its health condition
# But for now, this allows us to break long pieces of text
# into shorter chunks

class FeatureSetBuilder:
    def __init__(self, label):
        self.features = []
        self.label = label


    def add(self, input_feature):
        self.features.append(input_feature)

    def resize(self, num_sections, func):
        # if the num sections isn't maxed we either need to pad out or cut down.

        while len(self.features) < num_sections:
            self.features.append(func)

        # Handle the case where we have too many sections - cut at the head
        if len(self.features) > num_sections:
            self.features = self.features[-num_sections:]

    #returns all features
    def get(self):
        return self.features

    def get_feature(self, feature_index):
        return self.features[feature_index]

    def get_label(self):
        return self.label