i `dZddlZddlmZmZmZmZejeZ GddZ y)zUser simulator for evaluation.N)DictListSetAnyczeZdZdZ d deedeedzfdZdeeefde fdZ d e eeefdee e ffd Z y) UserSimulatorz-Simulates a user with predefined preferences.Nliked_featuresdisliked_featuresc||_|xs t|_tj d|jd|jy)zInitialize user simulator. Args: liked_features: Set of feature types the user likes disliked_features: Set of feature types the user dislikes (optional) z!UserSimulator initialized: liked=z , disliked=N)likedsetdislikedloggerinfo)selfr r s K/home/ubuntu/codebase/yexijia/保研/colocation_mvp/experiment/simulator.py__init__zUserSimulator.__init__ s;$ )2SU  7 |;t}}o^_patternreturnc|jdg}t|tr|jd}t fd|D}|syt fd|D}|r|sd}|Sd}|S)aLabel a pattern based on ground truth preferences. A pattern is liked iff: - contains at least one liked feature - does not contain any disliked feature If a pattern does not contain any liked feature, it is marked as not interested (0). Args: pattern: Pattern dictionary with 'pattern' key containing list of features Returns: 1 if pattern is liked, 0 if not interested or disliked r,c3:K|]}|jvywN)r .0featurers r z.UserSimulator.label_pattern../sN=M'4::-=Mrc3:K|]}|jvywr)rrs rrz.UserSimulator.label_pattern..6sTCS7dmm3CSr)get isinstancestrsplitany)rrpattern_features has_liked has_dislikedlabels` r label_patternzUserSimulator.label_patterns|#;;y"5 & ,/55c: N=MNN TCSTT   <= rpatternsci}t|D]\}}|j|||<td|jD}tj d|dt |d|S)zBuild ground truth labels for all patterns. Args: patterns: List of pattern dictionaries Returns: Dictionary mapping pattern index to label (0 or 1) c3,K|] }|dk(s dyw)r!N)rvs rrz3UserSimulator.build_ground_truth..IsE%:a1f!%:s zBuilt ground truth: /z patterns liked) enumerater+sumvaluesrrlen)rr, ground_truthidxr liked_counts rbuild_ground_truthz UserSimulator.build_ground_truth<sr %h/LC $ 2 27 ;L 0E\%8%8%:EE  *;-qXWXrr) __name__ __module__ __qualname____doc__rr$rrrintr+rr9r/rrrr st7 .2 `C `s8d? `T#s(^B4S#X+?DcNrr) r=loggingtypingrrrr getLoggerr:rrr/rrrBs/$''   8 $BBr