From 943d543fe9a465df1c81a31a6acf970dedfecf44 Mon Sep 17 00:00:00 2001
From: scastlara <s.cast.lara@gmail.com>
Date: Mon, 15 Oct 2018 11:56:55 +0200
Subject: [PATCH] Votes normalized.

---
 ppaxe/core.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/ppaxe/core.py b/ppaxe/core.py
index 699720a..d2c71c6 100644
--- a/ppaxe/core.py
+++ b/ppaxe/core.py
@@ -979,9 +979,21 @@ def predict(self):
         self.votes = round(pred[0], 3)
         if pred >= 0.55:
             self.label = True
+            self._normalize_pred()
         else:
             self.label = False
 
+
+    def _normalize_pred(self):
+        '''
+        Changes scale of votes in self.votes from 0.55-1 to 0-1.
+        '''
+        before = self.votes
+        self.votes = (self.votes-0.55)/(1-0.55)
+        self.votes = round(self.votes, 3)
+        print("B: %s A: %s\n" % (before, self.votes))
+
+
     def to_html(self):
         '''
         Transforms candidate to html with only involved proteins tagged and only