New translations synthesized_graphs_num_el.py (French)

lang/fr/gklearn/experiments/papers/PRL_2020/synthesized_graphs_num_el.py

@@ -0,0 +1,63 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Created on Mon Sep 21 10:34:26 2020
+
+@author: ljia
+"""
+from utils import Graph_Kernel_List_ESym, compute_graph_kernel
+import logging
+
+
+def generate_graphs(num_el_alp):
+	from gklearn.utils.graph_synthesizer import GraphSynthesizer
+	gsyzer = GraphSynthesizer()
+	graphs = gsyzer.unified_graphs(num_graphs=100, num_nodes=20, num_edges=40, num_node_labels=0, num_edge_labels=num_el_alp, seed=None, directed=False)
+	return graphs
+
+
+def xp_synthesied_graphs_num_edge_label_alphabet():
+		
+	# Run and save.
+	import pickle
+	import os
+	save_dir = 'outputs/synthesized_graphs_num_edge_label_alphabet/'
+	if not os.path.exists(save_dir):
+		os.makedirs(save_dir)
+
+	run_times = {}
+	
+	for kernel_name in Graph_Kernel_List_ESym:
+		print()
+		print('Kernel:', kernel_name)
+		
+		run_times[kernel_name] = []
+		for num_el_alp in [0, 4, 8, 12, 16, 20, 24, 28, 32, 36, 40]:
+			print()
+			print('Number of edge label alphabet:', num_el_alp)
+			
+			# Generate graphs.
+			graphs = generate_graphs(num_el_alp)
+
+			# Compute Gram matrix.
+			run_time = 'error'
+			try:
+				gram_matrix, run_time = compute_graph_kernel(graphs, kernel_name, n_jobs=1)
+			except Exception as exp:
+				print('An exception occured when running this experiment:')
+				LOG_FILENAME = save_dir + 'error.txt'
+				logging.basicConfig(filename=LOG_FILENAME, level=logging.DEBUG)
+				logging.exception('')
+				print(repr(exp))
+			run_times[kernel_name].append(run_time)
+			
+			pickle.dump(run_time, open(save_dir + 'run_time.' + kernel_name + '.' + str(num_el_alp) + '.pkl', 'wb'))
+		
+	# Save all.	
+	pickle.dump(run_times, open(save_dir + 'run_times.pkl', 'wb'))	
+	
+	return
+
+
+if __name__ == '__main__':
+	xp_synthesied_graphs_num_edge_label_alphabet()