Update app.py

app.py

@@ -43,7 +43,8 @@ class CRF_POS_Tagger:
         self.y_test = self.y[self.split:]
         self.crf_model = sklearn_crfsuite.CRF(algorithm='lbfgs', c1=0.1, c2=0.1, max_iterations=100, all_possible_transitions=True)
         if train:
-            self.train() 
+            data = list(zip(self.X_train, self.y_train))
+            self.train(data) 
 
     def word_splitter(self, word):
         prefix = ""
@@ -149,8 +150,8 @@ class CRF_POS_Tagger:
         
         return features
 
-    def train(self, data=None):
-        X_train, y_train = zip(*data) if data else self.X_train, self.y_train
+    def train(self, data):
+        X_train, y_train = zip(*data)
         self.crf_model.fit(X_train, y_train)
     
     def predict(self, X_test):
@@ -164,7 +165,7 @@ class CRF_POS_Tagger:
         return metrics.flat_accuracy_score(y_test, y_pred)
 
     def cross_validation(self):
-
+        validator = CRF_POS_Tagger()
         data = list(zip(self.X, self.y))  
         accuracies = []
         for i in range(5):
@@ -172,8 +173,8 @@ class CRF_POS_Tagger:
             n2 = int((i + 1) / 5.0 * len(data))
             test_data = data[n1:n2]
             train_data = data[:n1] + data[n2:]
-            self.train(train_data)
-            acc = self.accuracy(test_data)
+            validator.train(train_data)
+            acc = validator.accuracy(test_data)
             accuracies.append(acc)
         return accuracies, sum(accuracies) / 5.0
 
@@ -230,15 +231,14 @@ class CRF_POS_Tagger:
         return output
 
 
-# validator = CRF_POS_Tagger()
-# accuracies, avg_accuracy = validator.cross_validation()
-# print(f"Cross-Validation Accuracies: {accuracies}")
-# print(f"Average Accuracy: {avg_accuracy}")
+tagger = CRF_POS_Tagger(True)
+accuracies, avg_accuracy = tagger.cross_validation()
+print(f"Cross-Validation Accuracies: {accuracies}")
+print(f"Average Accuracy: {avg_accuracy}")
 
-# conf_matrix = tagger.con_matrix()
-# print(tagger.per_pos_accuracy(conf_matrix))
+conf_matrix = tagger.con_matrix()
+print(tagger.per_pos_accuracy(conf_matrix))
 
-tagger = CRF_POS_Tagger(True)
 interface = gr.Interface(fn = tagger.tagging, 
                             inputs = gr.Textbox(
                             label="Input Sentence",