plots.py

import matplotlib.patches as mpatches
import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np
import time

from auctions import *
from approximations import *
from simulations import *

def revenue_plot(n_bidders, n_items, k, n_trials):
	step_size = 0.3
	vcg_revenue = []
	gmsma_revenue = []
	vcg_runtime = []
	gmsma_runtime = []
	x = []
	for r in np.arange(0, 1, step_size):
		print(r)
		average_vcg_revenue = 0
		average_gmsma_revenue = 0
		average_vcg_runtime = 0
		average_gmsma_runtime = 0
		for i in range(n_trials):
			sim = MultiMindedSimulator(n_bidders, n_items, True, None, k, 2, r)
			sim2 = MultiMindedSimulator(n_bidders, n_items, False, additive_valuation_approximation, k, 2, r)
			t1 = time.time()
			result = sim.simulate()
			average_vcg_revenue += sum(price for outcome, price in result)
			average_vcg_runtime += time.time() - t1
			t2 = time.time()
			average_gmsma_revenue += sum(price for outcome, price in sim2.simulate())
			average_gmsma_runtime += time.time() - t2
		average_gmsma_revenue /= n_trials
		average_vcg_revenue /= n_trials
		average_gmsma_runtime /= n_trials
		average_vcg_runtime /= n_trials
		vcg_revenue.append(average_vcg_revenue)
		gmsma_revenue.append(average_gmsma_revenue)
		vcg_runtime.append(average_vcg_runtime)
		gmsma_runtime.append(average_gmsma_runtime)
	plt.plot(np.arange(0, 1, step_size), gmsma_runtime)
	plt.plot(np.arange(0, 1, step_size), vcg_runtime)
	plt.legend(['GM-SMA', 'VCG'])
	plt.xlabel('Proportion of Single-Minded Bidders')
	plt.ylabel('Protocol Execution Time (Seconds)')
	plt.title('VCG and GM-SMA Execution Time Comparison')
	plt.savefig('plots/runtime2.png')
	plt.clf()
	plt.plot(np.arange(0, 1, step_size), gmsma_revenue)
	plt.plot(np.arange(0, 1, step_size), vcg_revenue)
	plt.legend(['GM-SMA', 'VCG'])
	plt.xlabel('Proportion of Single-Minded Bidders')
	plt.ylabel('Revenue')
	plt.title('VCG and GM-SMA Revenue Comparison')
	plt.savefig('plots/revenue2.png')
	plt.show()

if __name__ == '__main__':
	random.seed(42)
	revenue_plot(6, 4, 2, 1000)