Commit d9e0b981 authored by Devanshu Agrawal's avatar Devanshu Agrawal
Browse files

Update report generation script.

parent 735fe337

torch_experiments/report/main.py

+10 −11
Original line number Diff line number Diff line
@@ -150,22 +150,21 @@ def tex(dataset, dataset_name, lc_depths, lc_widths, lod_widths=None): 




if __name__ == "__main__":

	datasets = ["boston", "iris", "rings"]

	datasets = ["boston", "iris_iters2000", "rings"]

	dataset_names = ["Boston housing prices", "Iris", "Rings"]



	for (dataset, dataset_name) in zip(datasets, dataset_names):

		hm_depths = list(range(1, 13))  # + [10*i for i in range(1, 10)]

		if dataset == "boston":

			hm_widths = [1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 0]

		else:

	hm_depths = list(range(1, 21))

	hm_widths = [1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 0]



		lc_depths = [1, 6, 12]  # [1, 10, 50, 90]

		if dataset == "boston":

	lc_depths = [1, 6, 12]

	lc_widths = [1, 32, 1024, 0]



	for (dataset, dataset_name) in zip(datasets, dataset_names):

		if dataset == "boston":

			omit_iters = 0

		else:

			lc_widths = [1, 32, 1024, 0]

		elif dataset == "iris_iters2000":

			omit_iters = 100

		elif dataset == "rings":

			omit_iters = 10



		heat_maps(dataset, hm_depths, hm_widths, save_csv=True)