Пример #1
0
def visual_test():
	print "===================="
	print "Visual Test: Small Disc"
	print "===================="
	sdisc = 'sdisc.txt'
	build_disc_small(sdisc)

	print "Original disc:"
	g1 = Graph(load_graph(sdisc))
	graph = Graph(load_graph(sdisc))
	output_graph(graph)

	print "Corrupted disc:"
	corrupt_image(graph, 0.1)
	output_graph(graph)

	print "Reconstructed disc:"
	theta = 0.5
	t = graph.loopy_bp(theta)
	output_graph(graph)
	print "Accuracy:", compare_graphs(g1, graph)
	print "Total iterations:", t
	print "===================="

	print "===================="
	print "Visual Test: Disc"
	print "===================="
	disc = 'disc.txt'
	build_disc(disc)

	print "Original disc:"
	g1 = Graph(load_graph(disc))
	graph = Graph(load_graph(disc))
	output_graph(graph)

	print "Corrupted disc:"
	corrupt_image(graph, 0.1)
	output_graph(graph)

	print "Reconstructed disc:"
	theta = 0.5
	t = graph.loopy_bp(theta)
	output_graph(graph)
	print "Accuracy:", compare_graphs(g1, graph)
	print "Total iterations:", t
	print "===================="
Пример #2
0
def parameter_test(mu, theta):
	disc = 'disc.txt'
	build_disc(disc)
	print "===================="
	print "Mu:", mu
	print "Theta:", theta
	g1 = Graph(load_graph(disc))
	graph = Graph(load_graph(disc))
	corrupt_image(graph, mu)
	t = graph.loopy_bp(theta)
	print "Accuracy:", compare_graphs(g1, graph)
	print "Total iterations:", t