Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
| 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):
|