segregation
index
/home/gcallah/TandonDevOps/IndraABM/models/segregation.py

Thomas Schelling's famous model of neighborhood segregation.

 
Modules
       
lib.actions
lib.model
random

 
Classes
       
lib.model.Model(builtins.object)
Segregation

 
class Segregation(lib.model.Model)
    Segregation(model_nm='model', props=None, grp_struct={'def_grp': {'mbr_creator': <function create_agent at 0x7c8bb5aeb790>, 'grp_action': None, 'mbr_action': <function def_action at 0x7c8bb5aeb8b0>, 'num_mbrs': 1, 'num_mbrs_prop': None, 'color': 'blue'}, 'red_grp': {'mbr_creator': <function create_agent at 0x7c8bb5aeb790>, 'grp_action': None, 'mbr_action': <function def_action at 0x7c8bb5aeb8b0>, 'num_mbrs': 1, 'num_mbrs_prop': None, 'color': 'red'}}, env_action=None, random_placing=True, serial_obj=None, exec_key=None, create_for_test=False)
 
Thomas Schelling's famous model of neighborhood segregation.
 
 
Method resolution order:
Segregation
lib.model.Model
builtins.object

Methods defined here:
handle_props(self, props)
A generic parameter handling method.
We get height and width here, since so many models use them.

Methods inherited from lib.model.Model:
__init__(self, model_nm='model', props=None, grp_struct={'def_grp': {'mbr_creator': <function create_agent at 0x7c8bb5aeb790>, 'grp_action': None, 'mbr_action': <function def_action at 0x7c8bb5aeb8b0>, 'num_mbrs': 1, 'num_mbrs_prop': None, 'color': 'blue'}, 'red_grp': {'mbr_creator': <function create_agent at 0x7c8bb5aeb790>, 'grp_action': None, 'mbr_action': <function def_action at 0x7c8bb5aeb8b0>, 'num_mbrs': 1, 'num_mbrs_prop': None, 'color': 'red'}}, env_action=None, random_placing=True, serial_obj=None, exec_key=None, create_for_test=False)
Initialize self.  See help(type(self)) for accurate signature.
__repr__(self)
This returns a JSON representation of the model.
__str__(self)
Return str(self).
add_child(self, group)
Put a child agent in the womb.
group: which group will add new agent
The womb should move up into model eventually.
add_switch(self, agent_nm, from_grp_nm, to_grp_nm)
Switch agent from 1 group to another.
The agent and groups should be passed by name.
bar_graph(self)
collect_stats(self)
create_anew(self, model_nm, props, grp_struct, exec_key, env_action, random_placing, create_for_test=False)
Create the model for the first time.
create_env(self, env_action=None, random_placing=True)
Override this method to create a unique env...
but this one will already set the model name and add
the groups.
create_from_serial_obj(self, serial_obj)
Restore the model from its serialized version.
create_groups(self)
Override this method in your model to create all of your groups.
In general, you shouldn't need to: fill in the grp_struct instead.
create_pop_hist(self)
There are several methods that still (like in V2) reside in
Env, but which we mean to move to Model. So we provide an interface to
them here so when we move them other code won't break.
`create_pop_hist()` is such a method.
create_user(self)
This will create a user of the correct type.
from_json(self, jrep)
This method restores a model from its JSON rep.
get_locations(self)
get_periods(self)
get_pop_hist(self)
get_prop(self, prop_nm, default=None)
Have a way to get a prop through the model to hide props structure.
get_user_msgs(self)
handle_args(self)
handle_switches(self)
This will actually process the pending switches.
handle_womb(self)
This method adds new agents from the womb.
The womb should move up into model eventually.
is_api_user(self)
is_test_user(self)
line_graph(self)
pending_switches(self)
How many switches are there to execute?
rpt_census(self, acts, moves)
This is the default census report.
Right now, `acts` is not used: do we need it?
Return: a string saying what happened in a period.
rpt_stats(self)
This is a "wrap up" report on the results of a model run.
Each model can do what it wants here.
perhaps will take an iterator object?
a file?
Function takes in a CSV formatted string from function
collect_stats() and writes it to a csv file.
 
Note: added logic so func will not write to stdout
rpt_switches(self)
Generate a string to report our switches.
run(self, periods=None)
This method runs the model. If `periods` is not None,
it will run it for that many periods. Otherwise, on
a terminal, it will display the menu.
Return: 0 if run was fine.
runN(self, periods=10)
Run our model for N periods.
Return the total number of actions taken.
run_batch(self, runs, steps)
Run our model for N periods X steps.
Writes the period specific model statistics to a CSV file.
Files are saved as input filename-[integer-counter].csv
Returns the total number of actions taken.
scatter_plot(self)
to_json(self)
This method generates the JSON representation for this model.
update_pop_hist(self)
This method records our populations each period.

Data descriptors inherited from lib.model.Model:
__dict__
dictionary for instance variables (if defined)
__weakref__
list of weak references to the object (if defined)

 
Functions
       
agent_action(agent, **kwargs)
This is what agents do each turn of the model.
create_model(serial_obj=None, props=None, create_for_test=False, use_exec_key=None)
This is for the sake of the API server:
env_favorable(hood_ratio, my_tolerance)
Is the environment to our agent's liking or not??
get_tolerance(default_tolerance, sigma)
`tolerance` measures how *little* of one's own group one will
tolerate being among.
main()

 
Data
        BLUE_AGENTS = 'Blue agents'
BLUE_GRP_IDX = 0
DEF_CITY_DIM = 40
DEF_HOOD_SIZE = 1
DEF_SIGMA = 0.2
DEF_TOLERANCE = 0.5
DEVIATION = 'deviation'
GRP_INDEX = 'grp_index'
HOOD_SIZE = 4
MAX_TOL = 0.9
MIN_TOL = 0.1
MODEL_NAME = 'segregation'
NOT_ZERO = 0.001
NUM_BLUE = 250
NUM_RED = 250
RED_AGENTS = 'Red agents'
RED_GRP_IDX = 1
TOLERANCE = 'tolerance'
group_names = ['Blue agents', 'Red agents']
hood_size = None
opp_group = None
segregation_grps = {'blue_group': {'color': 'blue', 'grp_action': None, 'mbr_action': <function agent_action>, 'num_mbrs': 250, 'num_mbrs_prop': 'num_blue'}, 'red_group': {'color': 'red', 'grp_action': None, 'mbr_action': <function agent_action>, 'num_mbrs': 250, 'num_mbrs_prop': 'num_red'}}