HaoYizhi/IIabm
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

firm request and accept preference to 2 level

Browse Source
This commit is contained in:
HaoYizhi
2023-06-11 16:09:30 +08:00
parent daab7af6ee
commit 67aa573154
8 changed files with 61 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

96
firm.py
View File

@@ -24,9 +24,9 @@ class FirmAgent(ap.Agent):
# para
self.flt_crit_supplier = float(self.p.crit_supplier)
self.flt_firm_req_prf_size = float(self.p.firm_req_prf_size)
self.is_firm_req_prf_size = bool(self.p.firm_req_prf_size)
self.is_firm_req_prf_conn = bool(self.p.firm_req_prf_conn)
self.flt_firm_acc_prf_size = float(self.p.firm_acc_prf_size)
self.is_firm_acc_prf_size = bool(self.p.firm_acc_prf_size)
self.is_firm_acc_prf_conn = bool(self.p.firm_acc_prf_conn)
self.flt_diff_new_conn = float(self.p.diff_new_conn)
@@ -97,27 +97,34 @@ class FirmAgent(ap.Agent):
if self.code in lst_adj_firm:
lst_firm_connect.append(firm)
if len(lst_firm_connect) == 0:
# select based on size
lst_size_damp = \
[size ** self.flt_firm_req_prf_size for size in
self.dct_cand_alt_supply_up_prod_removed[
product].revenue_log]
lst_prob = [size_damp / sum(lst_size_damp)
for size_damp in lst_size_damp]
select_alt_supply = self.model.nprandom.choice(
self.dct_cand_alt_supply_up_prod_removed[product],
p=lst_prob)
# select based on size or not
if self.is_firm_req_prf_size:
lst_size = \
[size for size in
self.dct_cand_alt_supply_up_prod_removed[
product].revenue_log]
lst_prob = [size / sum(lst_size)
for size in lst_size]
select_alt_supply = self.model.nprandom.choice(
self.dct_cand_alt_supply_up_prod_removed[product],
p=lst_prob)
else:
select_alt_supply = self.model.nprandom.choice(
self.dct_cand_alt_supply_up_prod_removed[product])
elif len(lst_firm_connect) > 0:
# select based on size of firm that has connection
lst_firm_size_damp = \
[firm.revenue_log ** self.flt_firm_acc_prf_size
for firm in lst_firm_connect]
lst_prob = \
[size_damp / sum(lst_firm_size_damp)
for size_damp in lst_firm_size_damp]
select_alt_supply = \
self.model.nprandom.choice(lst_firm_connect,
p=lst_prob)
# select based on size of connected firm or not
if self.is_firm_req_prf_size:
lst_firm_size = \
[firm.revenue_log for firm in lst_firm_connect]
lst_prob = \
[size / sum(lst_firm_size)
for size in lst_firm_size]
select_alt_supply = \
self.model.nprandom.choice(lst_firm_connect,
p=lst_prob)
else:
select_alt_supply = \
self.model.nprandom.choice(lst_firm_connect)
# print(
# f"{self.name} selct alt supply for {product.code} "
# f"from {select_alt_supply.name}"
@@ -174,27 +181,34 @@ class FirmAgent(ap.Agent):
if self.code in lst_adj_firm:
lst_firm_connect.append(firm)
if len(lst_firm_connect) == 0:
# handling based on size
lst_firm_size_damp = \
[firm.revenue_log ** self.flt_firm_acc_prf_size
for firm in lst_firm]
lst_prob = \
[size_damp / sum(lst_firm_size_damp)
for size_damp in lst_firm_size_damp]
select_customer = \
self.model.nprandom.choice(lst_firm, p=lst_prob)
# handling based on size or not
if self.is_firm_acc_prf_size:
lst_firm_size = \
[firm.revenue_log for firm in lst_firm]
lst_prob = \
[size / sum(lst_firm_size)
for size in lst_firm_size]
select_customer = \
self.model.nprandom.choice(lst_firm,
p=lst_prob)
else:
select_customer = \
self.model.nprandom.choice(lst_firm)
self.accept_request(select_customer, product)
elif len(lst_firm_connect) > 0:
# handling based on size of firm that has connection
lst_firm_size_damp = \
[firm.revenue_log ** self.flt_firm_acc_prf_size
for firm in lst_firm_connect]
lst_prob = \
[size_damp / sum(lst_firm_size_damp)
for size_damp in lst_firm_size_damp]
select_customer = \
self.model.nprandom.choice(lst_firm_connect,
p=lst_prob)
# handling based on size of connected firm or not
if self.is_firm_acc_prf_size:
lst_firm_size = \
[firm.revenue_log for firm in lst_firm_connect]
lst_prob = \
[size / sum(lst_firm_size)
for size in lst_firm_size]
select_customer = \
self.model.nprandom.choice(lst_firm_connect,
p=lst_prob)
else:
select_customer = \
self.model.nprandom.choice(lst_firm_connect)
self.accept_request(select_customer, product)
def accept_request(self, down_firm, product):