Merge branch '0.0.1' of https://github.com/Anandhex/ReachOut into 0.0.1

Anandhex · Apr 19, 2020 · 8854969 · 8854969
2 parents ab27fca + d909570
commit 8854969
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/ml/ml_source.py b/ml/ml_source.py
@@ -5,6 +5,7 @@
 from sklearn.cluster import KMeans
 import json,re
 import random
+import copy
 
 app = Flask(__name__)
 
@@ -139,7 +140,11 @@ def predict_interest():
     for rule in association_rules:
         if rule["Antecedent"] == interests:
             return jsonify({"interests":rule["Consequent"]})
-    return jsonify({"interests":"Found Nothing"})
+
+    inte = copy.deepcopy(list(interest_id_dict.keys()))
+    random.shuffle(inte)
+
+    return jsonify({"interests":inte[:3]})
 
 
 if __name__ == "__main__":