Update qbmodel.py

qbmodel.py

@@ -170,7 +170,7 @@ class QuizBowlModel:
         #max_confidence, max_answer = selected_model.predict(input_data)
         #print("Prediction for category", self.categories[category], ":", max_answer, "with confidence", max_confidence)
         
-        return (np.tanh(max_confidence), max_answer)
+        return (self.confidence_eq(np.tanh(max_confidence)), max_answer)
         
     def evaluate(self, input_data):
         correct = 0
@@ -199,6 +199,18 @@ class QuizBowlModel:
           accuracy = correct/len(input_data)
           
           return accuracy
+
+    def confidence_eq(self,x):
+        
+        if(x < 0.5):
+            return 0
+        elif(x < 0.6):
+            return 2*x - 1
+        elif(x < 0.7):
+            return 4*x - 2.2
+        else:
+            return min(1, 1.5*(x + 0.1)**2 - 0.36)
+