Add tag encoder to FFNN (#15)

- Add tag encoder to FFNN (c7e0f49f1c0e4e5f509d7fbb51fa4b355a675834)


Co-authored-by: H Wilks <[email protected]>

customFunctions.py

@@ -466,6 +466,9 @@ class FeedforwardNN(BaseEstimator, ClassifierMixin):
                 flattened_pred = predicted_tags.view(-1)
                 predictions.append(flattened_pred.cpu().numpy())
 
+        tag_encoder = LabelEncoder()
+        tag_encoder.fit(['B-AC', 'O', 'B-LF', 'I-LF'])
+
         str_pred = []
         for sentence in predictions:
             str_sentence = tag_encoder.inverse_transform(sentence)