Update trend classification algorithm

CDCgov · May 31, 2023 · 4150b01 · 4150b01
1 parent 1ef5a9e
commit 4150b01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pynssp/detectors/trend.py b/pynssp/detectors/trend.py
@@ -91,7 +91,7 @@ def classify_trend(df, t='date', data_count='dataCount', all_count='allCount', B
         >>> df = pd.DataFrame({
         ...     "date": pd.date_range("2020-01-01", "2020-12-31"),
         ...     "dataCount": np.random.randint(0, 101, size=366),
-        ...     "allCount": np.random.randint(0, 101, size=366)
+        ...     "allCount": np.random.randint(101, 500, size=366)
         ... })
         >>> df_trend = classify_trend(df)
         >>> df_trend.head()