Spaces:

keras-io
/

timeseries_transformer_classification

Runtime error

joheras commited on Feb 3, 2022

Commit

114644e

1 Parent(s): fb7e6f8

app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ def detect_issue(file):
   preds = model.predict(df)
   result = []
   for i,pred in enumerate(preds):
-    result.append(['Sample ' + str(i+1), np.argmax(pred),pred[np.argmax(pred)]]
   return pd.DataFrame(result,header=['Sample','class','confidence'])

   preds = model.predict(df)
   result = []
   for i,pred in enumerate(preds):
+    result.append(['Sample ' + str(i+1), np.argmax(pred),pred[np.argmax(pred)]])
   return pd.DataFrame(result,header=['Sample','class','confidence'])