Spaces:
Runtime error
Runtime error
File size: 3,895 Bytes
82a7a28 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
import random
import pandas as pd
from tinytroupe.agent import TinyPerson
class ABRandomizer():
def __init__(self, real_name_1="control", real_name_2="treatment",
blind_name_a="A", blind_name_b="B",
passtrough_name=[],
random_seed=42):
"""
An utility class to randomize between two options, and de-randomize later.
The choices are stored in a dictionary, with the index of the item as the key.
The real names are the names of the options as they are in the data, and the blind names
are the names of the options as they are presented to the user. Finally, the passtrough names
are names that are not randomized, but are always returned as-is.
Args:
real_name_1 (str): the name of the first option
real_name_2 (str): the name of the second option
blind_name_a (str): the name of the first option as seen by the user
blind_name_b (str): the name of the second option as seen by the user
passtrough_name (list): a list of names that should not be randomized and are always
returned as-is.
random_seed (int): the random seed to use
"""
self.choices = {}
self.real_name_1 = real_name_1
self.real_name_2 = real_name_2
self.blind_name_a = blind_name_a
self.blind_name_b = blind_name_b
self.passtrough_name = passtrough_name
self.random_seed = random_seed
def randomize(self, i, a, b):
"""
Randomly switch between a and b, and return the choices.
Store whether the a and b were switched or not for item i, to be able to
de-randomize later.
Args:
i (int): index of the item
a (str): first choice
b (str): second choice
"""
# use the seed
if random.Random(self.random_seed).random() < 0.5:
self.choices[i] = (0, 1)
return a, b
else:
self.choices[i] = (1, 0)
return b, a
def derandomize(self, i, a, b):
"""
De-randomize the choices for item i, and return the choices.
Args:
i (int): index of the item
a (str): first choice
b (str): second choice
"""
if self.choices[i] == (0, 1):
return a, b
elif self.choices[i] == (1, 0):
return b, a
else:
raise Exception(f"No randomization found for item {i}")
def derandomize_name(self, i, blind_name):
"""
Decode the choice made by the user, and return the choice.
Args:
i (int): index of the item
choice_name (str): the choice made by the user
"""
# was the choice i randomized?
if self.choices[i] == (0, 1):
# no, so return the choice
if blind_name == self.blind_name_a:
return self.real_name_1
elif blind_name == self.blind_name_b:
return self.real_name_2
elif blind_name in self.passtrough_name:
return blind_name
else:
raise Exception(f"Choice '{blind_name}' not recognized")
elif self.choices[i] == (1, 0):
# yes, it was randomized, so return the opposite choice
if blind_name == self.blind_name_a:
return self.real_name_2
elif blind_name == self.blind_name_b:
return self.real_name_1
elif blind_name in self.passtrough_name:
return blind_name
else:
raise Exception(f"Choice '{blind_name}' not recognized")
else:
raise Exception(f"No randomization found for item {i}")
|