Browse Source

Add parameter match_wifi_num_crit

Xi Ruoyao 5 years ago
parent
commit
75f59daf5d
2 changed files with 6 additions and 2 deletions
  1. 3 1
      cosine.py
  2. 3 1
      otsuka.py

+ 3 - 1
cosine.py

@@ -1,5 +1,7 @@
 #!/usr/bin/env python3
 
+match_wifi_num_crit = 2
+
 def cosine_sim(known_point, new_point):
     from math import sqrt
     up, d1, d2 = 0.0, 0.0, 0.0
@@ -10,7 +12,7 @@ def cosine_sim(known_point, new_point):
             up += a.wifi_snr[key] * b.wifi_snr[key]
     for key in b.wifi_snr:
         d2 += b.wifi_snr[key] * b.wifi_snr[key]
-    if up == 0:
+    if up < match_wifi_num_crit:
         return None
     else:
         return up / sqrt(d1) / sqrt(d2)

+ 3 - 1
otsuka.py

@@ -1,5 +1,7 @@
 #!/usr/bin/env python3
 
+match_wifi_num_crit = 2
+
 # $K = \frac{|A \cap B|}{\sqrt{|A| \times |B|}}$
 def otsuka_sim(known_point, new_point):
     from math import sqrt
@@ -11,7 +13,7 @@ def otsuka_sim(known_point, new_point):
             up += 1
     for key in b.wifi_snr:
         d2 += 1
-    if up == 0:
+    if up < match_wifi_num_crit:
         return None
     else:
         return up / sqrt(d1) / sqrt(d2)