iKdZddlZddlmZmZmZddlmZmZm Z m Z m Z m Z m Z deeefdedefd Zdeeefd edefd Zd eedefd ZdededefdZy)zR Pure helpers for StatisticalAnalyzer (entropy, HHI, recency weight, confidence). N)DictAnyOptional)RECENCY_WEIGHT_7_DAYSRECENCY_WEIGHT_30_DAYSRECENCY_WEIGHT_60_DAYSRECENCY_WEIGHT_OLDERCONFIDENCE_NORMALIZERCONFIDENCE_HHI_OFFSETCONFIDENCE_HHI_FACTORcountstotalreturnc|dks|syd}|jD](}|dkr ||z }||tj|zz}*t|dS)z7Shannon entropy (base 2). total = sum(counts.values()).r)valuesmathlog2round)rrentcountps K/var/www/html/userprofiledev.eatanceapp.com/statistical_analysis/helpers.pyentropy_from_countsrs[ z C A:  EM q499Q< ! a=_totalc|sy|jDcgc] }|dkDs | }}t|dkryttfd|DdScc}w)zYHerfindahl-Hirschman Index. Uses only strictly positive counts; denominator is their sum.rrc3.K|] }|z dzyw)rN).0cts r z"hhi_from_counts..(s59aa!e\9sr)rsumr)rrr# positivesr$s @rhhi_from_countsr( sY "MMO5Oqq1uOI5 IAAv 5955q 99 6s AAlast_order_days_agocV|y|dkrtS|dkrtS|dkrtStS)z.Recency decay weight from last order days ago.r<)rrr r )r)s rweighted_recencyr.+s;"a$$b %%b %% r total_ordersrestaurant_hhic|dkrytj|t|tzzz}t t d|t z dS)z/Normalized confidence from order count and HHI.rrg?r)rlog1pr r rminr )r/r0raws rmath_confidencer58sFq **\ "&;~Pe?e&f gC S# 556 ::r)__doc__rtypingrrr constantsrrr r r r r strintfloatrr(r.r5r!rrr<s && S#X s u :DcN:C:E:  (3-  E  ;#;u;;r