Flexible framework for building agent-based modeling simulations.
- python 3
Install using pip (maybe be pip3 on some platforms):
pip install dworp
To install with the optional plotting capability:
pip install dworp[plot]
This will install dependencies required by dworp.plot
(currently only matplotlib).
Dworp defines basic interfaces for building simulations and provides some default components to support rapid creation of agent-based models.
An Agent
updates its state in the step()
function.
The update may depend on the environment, its neighbors, past history, or other features.
An agent has two optional functions init()
and complete()
.
The init()
function is called when an agent is added to the simulation.
The complete()
function is called at the end of a time step.
class MyAgent(dworp.Agent):
def step(self, now, env):
# ToDo add example here
pass
When agents are updating their state based on their neighbors' state, you may want to use a two stage update mechanism. In the first stage, the agents update their state privately so that their neighbors cannot see the new state. In the second stage, the agents make that state public to prepare for the next time step.
The Environment
captures all simulation state that does not live in the agents.
This includes serving as a container for network or spatial information for determining neighbors.
Time
drives the simulation and implements an iterator interface.
It can be fixed in length or infinite.
Time steps can be fixed in length or variable.
Time can be integer or floating point.
Agents can observe or interact with other agents based on neighborhoods. A neighborhood may be defined on a network using various graph frameworks like igraph or snap. A neighborhood can also be spatially defined on a grid or continuous space.
To stop the simulation when some condition is met, use a Terminator
.
The order that agents update and which agents update is determined by the Scheduler
.
Some basic schedulers are provided for round robin updates in random order or uniformly sampling.
An Observer
runs after each time step.
It is designed for capturing data for further processing.
It has access to the agents and the environment.
Multiple observers can be chained together using ChainedObserver
.
The Simulation
interface defines a single realization of an agent-based simulation.
Basic implementations for single stage and double stage updates are provided.
Usually, you will want to inherit from one of those to define your simulation.
Each component has its own logger:
self.logger.info("Agent {} set activity to {}".format(self.agent_id, self.activity))
The logging level can be controlled at the framework level:
logging.getLogger('dworp').setLevel(logging.WARN)
or at the individual component level:
logging.getLogger('dworp.agent').setLevel(logging.DEBUG)
For more information on log levels or log configuration, read the python logging docs.
The best way to learn the framework is by looking at the example models and their documentation.
Install the test requirements:
pip install -r test_requirements.txt
Then to run the tests (while in the base directory):
nosetests
This will run all the tests. To skip the plotting unit tests which require matplotlib, do this:
nosetests -a '!plot'
To get a report on unit test coverage:
nosetests --with-coverage --cover-package=dworp
To install in editable model so that changes to the framework are instantly reflected:
pip install -e .
The code mostly follows the PEP8 coding standard. If you are using PyCharm, it will highlight PEP8 issues. You can also manually run style checks with flake8 (pip install flake8):
flake8 dworp
The docstrings are using the Google standard.