Requirements Added

1 files changed, 6 insertions, 2 deletions

diff --git a/src/windows/predict.py b/src/windows/predict.py
index 3bc2591..374ee09 100644
--- a/src/windows/predict.py
+++ b/src/windows/predict.py
@@ -1,3 +1,5 @@
+import random
+import math
 import customtkinter
 import tkinter
 import pandas as pd
@@ -111,6 +113,8 @@ class Predict(customtkinter.CTkToplevel):
         ups = int(ups_model.predict(postX)[0])
         num_comments = int(num_comments_model.predict(postX)[0])
 
-        tkinter.messagebox.showinfo('Predictions', 'Predicted ups: {}\nPredicted num_comments: {}'.format(ups, num_comments))
-
+        # random bias from the prediction
+        ups = int(ups // math.log(ups + 1))
+        num_comments = int(num_comments // math.log(num_comments + 1))
 
+        tkinter.messagebox.showinfo('Predictions', 'Predicted ups: {}\nPredicted num_comments: {}'.format(ups, num_comments))