--- a
+++ b/medicalbert/datareader/FeatureSetBuilder.py
@@ -0,0 +1,36 @@
+# 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
\ No newline at end of file