From 6312ebe75af363ce711ce8c07a70830c076e4fd5 Mon Sep 17 00:00:00 2001 From: Wang Boyu Date: Mon, 2 Sep 2024 16:39:36 -0400 Subject: [PATCH] add model default params for agents-networks gis example --- gis/agents_and_networks/src/model/model.py | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) diff --git a/gis/agents_and_networks/src/model/model.py b/gis/agents_and_networks/src/model/model.py index e0b03114..15c0c297 100644 --- a/gis/agents_and_networks/src/model/model.py +++ b/gis/agents_and_networks/src/model/model.py @@ -1,5 +1,6 @@ import uuid from functools import partial +from pathlib import Path import geopandas as gpd import mesa @@ -13,6 +14,8 @@ from ..space.campus import Campus from ..space.road_network import CampusWalkway +script_directory = Path(__file__).resolve().parent + def get_time(model) -> pd.Timedelta: return pd.Timedelta(days=model.day, hours=model.hour, minutes=model.minute) @@ -59,14 +62,14 @@ class AgentsAndNetworks(mesa.Model): def __init__( self, - campus: str, - data_crs: str, - buildings_file: str, - walkway_file: str, - lakes_file: str, - rivers_file: str, - driveway_file: str, - num_commuters, + campus="ub", + data_crs="epsg:4326", + buildings_file=script_directory / "../../data/ub/UB_bld.zip", + walkway_file=script_directory / "../../data/ub/UB_walkway_line.zip", + lakes_file=script_directory / "../../data/ub/hydrop.zip", + rivers_file=script_directory / "../../data/ub/hydrol.zip", + driveway_file=script_directory / "../../data/data/ub/UB_Rds.zip", + num_commuters=50, commuter_min_friends=5, commuter_max_friends=10, commuter_happiness_increase=0.5,