X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=mlp.py;h=7ddb894dd7248a54861dad3e9ef9af585a2c6372;hb=546094cd63776f6d2a5d6ce1a0fa935a7b9dc956;hp=21a772c6331312abde4759db4d3250d6f474b36b;hpb=21558202134d649b29fb700481b15963e71b1d1f;p=agtree2dot.git diff --git a/mlp.py b/mlp.py index 21a772c..7ddb894 100755 --- a/mlp.py +++ b/mlp.py @@ -61,8 +61,18 @@ agtree2dot.save_dot(loss, print('Generated mlp.dot') try: - subprocess.check_call(['dot', 'mlp.dot', '-Lg', '-T', 'pdf', '-o', 'mlp.pdf' ]) + + fontname = 'Computer Modern' + fontsize = 12 + subprocess.check_call(['dot', 'mlp.dot', + '-Lg', + '-T', 'pdf', + '-Efontname=' + fontname, '-Efontsize=' + str(fontsize), + '-Nfontname=' + fontname, '-Nfontsize=' + str(fontsize), + '-o', 'mlp.pdf' ]) + except subprocess.CalledProcessError: + print('Calling the dot command failed. Is Graphviz installed?') sys.exit(1)