Mesa 3.0.0__py3-none-any.whl → 3.0.0a1__py3-none-any.whl

mesa/examples/advanced/epstein_civil_violence/app.py

@@ -1,73 +0,0 @@
-from mesa.examples.advanced.epstein_civil_violence.agents import (
-    Citizen,
-    CitizenState,
-    Cop,
-)
-from mesa.examples.advanced.epstein_civil_violence.model import EpsteinCivilViolence
-from mesa.visualization import (
-    Slider,
-    SolaraViz,
-    make_plot_component,
-    make_space_component,
-)
-
-COP_COLOR = "#000000"
-
-agent_colors = {
-    CitizenState.ACTIVE: "#FE6100",
-    CitizenState.QUIET: "#648FFF",
-    CitizenState.ARRESTED: "#808080",
-}
-
-
-def citizen_cop_portrayal(agent):
-    if agent is None:
-        return
-
-    portrayal = {
-        "size": 50,
-    }
-
-    if isinstance(agent, Citizen):
-        portrayal["color"] = agent_colors[agent.state]
-    elif isinstance(agent, Cop):
-        portrayal["color"] = COP_COLOR
-
-    return portrayal
-
-
-def post_process(ax):
-    ax.set_aspect("equal")
-    ax.set_xticks([])
-    ax.set_yticks([])
-    ax.get_figure().set_size_inches(10, 10)
-
-
-model_params = {
-    "height": 40,
-    "width": 40,
-    "citizen_density": Slider("Initial Agent Density", 0.7, 0.0, 0.9, 0.1),
-    "cop_density": Slider("Initial Cop Density", 0.04, 0.0, 0.1, 0.01),
-    "citizen_vision": Slider("Citizen Vision", 7, 1, 10, 1),
-    "cop_vision": Slider("Cop Vision", 7, 1, 10, 1),
-    "legitimacy": Slider("Government Legitimacy", 0.82, 0.0, 1, 0.01),
-    "max_jail_term": Slider("Max Jail Term", 30, 0, 50, 1),
-}
-
-space_component = make_space_component(
-    citizen_cop_portrayal, post_process=post_process, draw_grid=False
-)
-
-chart_component = make_plot_component(
-    {state.name.lower(): agent_colors[state] for state in CitizenState}
-)
-
-epstein_model = EpsteinCivilViolence()
-
-page = SolaraViz(
-    epstein_model,
-    components=[space_component, chart_component],
-    model_params=model_params,
-    name="Epstein Civil Violence",
-)
-page  # noqa

mesa/examples/advanced/epstein_civil_violence/model.py

@@ -1,114 +0,0 @@
-import mesa
-from mesa.examples.advanced.epstein_civil_violence.agents import (
-    Citizen,
-    CitizenState,
-    Cop,
-)
-
-
-class EpsteinCivilViolence(mesa.Model):
-    """
-    Model 1 from "Modeling civil violence: An agent-based computational
-    approach," by Joshua Epstein.
-    http://www.pnas.org/content/99/suppl_3/7243.full
-
-    Args:
-        height: grid height
-        width: grid width
-        citizen_density: approximate % of cells occupied by citizens.
-        cop_density: approximate % of cells occupied by cops.
-        citizen_vision: number of cells in each direction (N, S, E and W) that
-            citizen can inspect
-        cop_vision: number of cells in each direction (N, S, E and W) that cop
-            can inspect
-        legitimacy:  (L) citizens' perception of regime legitimacy, equal
-            across all citizens
-        max_jail_term: (J_max)
-        active_threshold: if (grievance - (risk_aversion * arrest_probability))
-            > threshold, citizen rebels
-        arrest_prob_constant: set to ensure agents make plausible arrest
-            probability estimates
-        movement: binary, whether agents try to move at step end
-        max_iters: model may not have a natural stopping point, so we set a
-            max.
-    """
-
-    def __init__(
-        self,
-        width=40,
-        height=40,
-        citizen_density=0.7,
-        cop_density=0.074,
-        citizen_vision=7,
-        cop_vision=7,
-        legitimacy=0.8,
-        max_jail_term=1000,
-        active_threshold=0.1,
-        arrest_prob_constant=2.3,
-        movement=True,
-        max_iters=1000,
-        seed=None,
-    ):
-        super().__init__(seed=seed)
-        self.movement = movement
-        self.max_iters = max_iters
-
-        self.grid = mesa.experimental.cell_space.OrthogonalVonNeumannGrid(
-            (width, height), capacity=1, torus=True, random=self.random
-        )
-
-        model_reporters = {
-            "active": CitizenState.ACTIVE.name,
-            "quiet": CitizenState.QUIET.name,
-            "arrested": CitizenState.ARRESTED.name,
-        }
-        agent_reporters = {
-            "jail_sentence": lambda a: getattr(a, "jail_sentence", None),
-            "arrest_probability": lambda a: getattr(a, "arrest_probability", None),
-        }
-        self.datacollector = mesa.DataCollector(
-            model_reporters=model_reporters, agent_reporters=agent_reporters
-        )
-        if cop_density + citizen_density > 1:
-            raise ValueError("Cop density + citizen density must be less than 1")
-
-        for cell in self.grid.all_cells:
-            klass = self.random.choices(
-                [Citizen, Cop, None],
-                cum_weights=[citizen_density, citizen_density + cop_density, 1],
-            )[0]
-
-            if klass == Cop:
-                cop = Cop(self, vision=cop_vision, max_jail_term=max_jail_term)
-                cop.move_to(cell)
-            elif klass == Citizen:
-                citizen = Citizen(
-                    self,
-                    regime_legitimacy=legitimacy,
-                    threshold=active_threshold,
-                    vision=citizen_vision,
-                    arrest_prob_constant=arrest_prob_constant,
-                )
-                citizen.move_to(cell)
-
-        self.running = True
-        self._update_counts()
-        self.datacollector.collect(self)
-
-    def step(self):
-        """
-        Advance the model by one step and collect data.
-        """
-        self.agents.shuffle_do("step")
-        self._update_counts()
-        self.datacollector.collect(self)
-
-        if self.steps > self.max_iters:
-            self.running = False
-
-    def _update_counts(self):
-        """Helper function for counting nr. of citizens in given state."""
-        counts = self.agents_by_type[Citizen].groupby("state").count()
-
-        for state in CitizenState:
-            setattr(self, state.name, counts.get(state, 0))

mesa/examples/advanced/pd_grid/Readme.md

@@ -1,43 +0,0 @@
-# Demographic Prisoner's Dilemma on a Grid
-
-## Summary
-
-The Demographic Prisoner's Dilemma is a family of variants on the classic two-player [Prisoner's Dilemma]. The model consists of agents, each with a strategy of either Cooperate or Defect. Each agent's payoff is based on its strategy and the strategies of its spatial neighbors. After each step of the model, the agents adopt the strategy of their neighbor with the highest total score.
-
-The model payoff table is:
-
-|               | Cooperate | Defect|
-|:-------------:|:---------:|:-----:|
-| **Cooperate** | 1, 1      | 0, D  |
-| **Defect**    | D, 0      | 0, 0  |
-
-Where *D* is the defection bonus, generally set higher than 1. In these runs, the defection bonus is set to $D=1.6$.
-
-The Demographic Prisoner's Dilemma demonstrates how simple rules can lead to the emergence of widespread cooperation, despite the Defection strategy dominating each individual interaction game. However, it is also interesting for another reason: it is known to be sensitive to the activation regime employed in it.
-
-## How to Run
-
-##### Web based model simulation
-
-To run the model interactively, run ``solara run app.py`` in this directory.
-
-##### Jupyter Notebook
-
-Launch the ``Demographic Prisoner's Dilemma Activation Schedule.ipynb`` notebook and run the code.
-
-## Files
-
-* ``agents.py``: contains the agent class.
-* ``model.py``: contains the model class; the model takes a ``activation_order`` string as an argument, which determines in which order agents are activated: Sequential, Random or Simultaneous.
-* ``app.py``: contains the interactive visualization server.
-* ``Demographic Prisoner's Dilemma Activation Schedule.ipynb``: Jupyter Notebook for running the scheduling experiment. This runs the model three times, one for each activation type, and demonstrates how the activation regime drives the model to different outcomes.
-
-## Further Reading
-
-This model is adapted from:
-
-Wilensky, U. (2002). NetLogo PD Basic Evolutionary model. http://ccl.northwestern.edu/netlogo/models/PDBasicEvolutionary. Center for Connected Learning and Computer-Based Modeling, Northwestern University, Evanston, IL.
-
-The Demographic Prisoner's Dilemma originates from:
-
-[Epstein, J. Zones of Cooperation in Demographic Prisoner's Dilemma. 1998.](http://citeseerx.ist.psu.edu/viewdoc/download?doi=10.1.1.8.8629&rep=rep1&type=pdf)

mesa/examples/advanced/pd_grid/agents.py

@@ -1,50 +0,0 @@
-from mesa.experimental.cell_space import CellAgent
-
-
-class PDAgent(CellAgent):
-    """Agent member of the iterated, spatial prisoner's dilemma model."""
-
-    def __init__(self, model, starting_move=None):
-        """
-        Create a new Prisoner's Dilemma agent.
-
-        Args:
-            model: model instance
-            starting_move: If provided, determines the agent's initial state:
-                           C(ooperating) or D(efecting). Otherwise, random.
-        """
-        super().__init__(model)
-        self.score = 0
-        if starting_move:
-            self.move = starting_move
-        else:
-            self.move = self.random.choice(["C", "D"])
-        self.next_move = None
-
-    @property
-    def is_cooroperating(self):
-        return self.move == "C"
-
-    def step(self):
-        """Get the best neighbor's move, and change own move accordingly
-        if better than own score."""
-
-        # neighbors = self.model.grid.get_neighbors(self.pos, True, include_center=True)
-        neighbors = [*list(self.cell.neighborhood.agents), self]
-        best_neighbor = max(neighbors, key=lambda a: a.score)
-        self.next_move = best_neighbor.move
-
-        if self.model.activation_order != "Simultaneous":
-            self.advance()
-
-    def advance(self):
-        self.move = self.next_move
-        self.score += self.increment_score()
-
-    def increment_score(self):
-        neighbors = self.cell.neighborhood.agents
-        if self.model.activation_order == "Simultaneous":
-            moves = [neighbor.next_move for neighbor in neighbors]
-        else:
-            moves = [neighbor.move for neighbor in neighbors]
-        return sum(self.model.payoff[(self.move, move)] for move in moves)