lbiester commited on
Commit
bec31a3
·
verified ·
1 Parent(s): 7983b2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def predict_sentiment(text, model):
41
  def plot():
42
  actual = ["pos", "pos", "neg", "neg", "pos"]
43
  pred = ["pos", "neg", "pos", "neg", "pos"]
44
- cm = confusion_matrix(y_test, predictions)
45
  disp = ConfusionMatrixDisplay(confusion_matrix=cm)
46
  disp.plot()
47
  return plt.gca()
 
41
  def plot():
42
  actual = ["pos", "pos", "neg", "neg", "pos"]
43
  pred = ["pos", "neg", "pos", "neg", "pos"]
44
+ cm = confusion_matrix(actual, pred)
45
  disp = ConfusionMatrixDisplay(confusion_matrix=cm)
46
  disp.plot()
47
  return plt.gca()