madhavkotecha commited on
Commit
89435c1
·
verified ·
1 Parent(s): 7124451

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -121,7 +121,7 @@ class NEI:
121
  title = 1 if w[0].isupper() else 0
122
  allcaps = 1 if w.isupper() else 0
123
  sw = 1 if w.lower() in SW else 0
124
- punct = 1 if w in PUNCT else 0
125
  return [title, allcaps, len(w), sw, punct, scaled_position]
126
 
127
  def create_data(self, data):
@@ -149,9 +149,8 @@ class NEI:
149
  _, X_val, y_val = self.create_data(val_data)
150
  X_val = self.scaler.transform(X_val)
151
  y_pred_val = self.model.predict(X_val)
152
-
153
  self.confusion_matrix(y_val,y_pred_val)
154
-
155
  self.performance(y_val,y_pred_val)
156
 
157
  def infer(self, sentence):
 
121
  title = 1 if w[0].isupper() else 0
122
  allcaps = 1 if w.isupper() else 0
123
  sw = 1 if w.lower() in SW else 0
124
+ punct = 1 if w in PUNCT else 0
125
  return [title, allcaps, len(w), sw, punct, scaled_position]
126
 
127
  def create_data(self, data):
 
149
  _, X_val, y_val = self.create_data(val_data)
150
  X_val = self.scaler.transform(X_val)
151
  y_pred_val = self.model.predict(X_val)
152
+ # print(classification_report(y_true=y_val, y_pred=y_pred_val))
153
  self.confusion_matrix(y_val,y_pred_val)
 
154
  self.performance(y_val,y_pred_val)
155
 
156
  def infer(self, sentence):