nes470
/

wiki-tfidf-logreg-qa

Question Answering

QA-umd-quizbowl

Model card Files Files and versions Community

nes470 commited on May 13

Commit

67bda20

•

1 Parent(s): d43a84b

Update LogRegBuzzer.py

Files changed (1) hide show

LogRegBuzzer.py +1 -1

LogRegBuzzer.py CHANGED Viewed

@@ -33,7 +33,7 @@ class LogisticRegressionBuzzer:
         prob_pred = self.model.predict_proba(X_formatted)
         print(prob_pred)
-        return (pred, prob_pred[0][1])

         prob_pred = self.model.predict_proba(X_formatted)
         print(prob_pred)
+        return (pred, float(pred[0]))