Filter results

results.py

@@ -1,10 +1,15 @@
+import numpy as np
 import pandas as pd
 
 from languages import en, ru
 
 
-def filter_results(results):
-    
+def filter_results(results : pd.DataFrame, top_k=0.95):
+    certs = results[certainty_init].to_numpy()
+    cum_certs = certs.cumsum()
+    do_no_show = cum_certs < top_k
+    first_not_to_show_id = np.argmin(do_no_show) + 1
+    results = results.iloc[:first_not_to_show_id]
     return results