Merge pull request #155 from TrafficCop/main

Made Nutrisense CGM Generator more reasonable

wearipedia/devices/nutrisense/cgm_gen.py

@@ -80,7 +80,9 @@ def gen_glucose(t, index, seed=0):
  X.append(x)
  Y.append(y)
 
- added = local_rng.normal(scale=1) * 10 + 0.01 * (160 / y)
+ added = sorted([-1.1, local_rng.normal(scale=1), 1.1])[1] * 10 + 0.01 * (
+ 160 / y
+ )
  if y < 75:
  added = abs(added)
  elif y > 135:
@@ -160,17 +162,7 @@ def gen_stats(Y, weekly=False):
  :rtype: dict
  """
 
- # if weekly, statistics not based on synthetic data Y, generate randomly
  if weekly:
- low = randint(70, 110)
- high = randint(low, 140)
- median = randint(low, high)
- std = randint(0, 10)
- q1 = randint(low, median)
- q3 = randint(median, high)
- timeWithinRange = randint(0, 100)
- avg = randint(low, high)
- else:
  low, high = min(Y), max(Y)
  median = np.median(Y)
  std = tstd(Y)
@@ -180,6 +172,15 @@ def gen_stats(Y, weekly=False):
  condition = greater & less
  timeWithinRange = float(len(np.extract(condition, Y)))
  avg = np.average(Y)
+ else:
+ low = randint(min(Y), 110)
+ high = randint(low, max(Y))
+ median = randint(low, high)
+ std = randint(0, 10)
+ q1 = randint(low, median)
+ q3 = randint(median, high)
+ timeWithinRange = randint(0, 100)
+ avg = randint(low, high)
 
  first = {"min": 70.0, "max": 140.0, "__typename": "Range"}
  second = {"min": low, "max": high, "__typename": "Range"}