X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=mlp.py;h=21d166bba9447a87315776cd741ad6d6f591e841;hb=765afb07c2781c92805461eb937d611fe03ca9bf;hp=963e7b7d55ac267f0beac10ec9f44f12506f9f93;hpb=f06a806854deeb2bb50324045a10b4a593927655;p=agtree2dot.git

diff --git a/mlp.py b/mlp.py
index 963e7b7..21d166b 100755
--- a/mlp.py
+++ b/mlp.py
@@ -22,7 +22,6 @@ import subprocess
 from torch import nn
 from torch.nn import functional as fn
 from torch import Tensor
-from torch.autograd import Variable
 from torch.nn import Module
 
 import agtree2dot
@@ -40,8 +39,8 @@ class MLP(Module):
         return x
 
 mlp = MLP(10, 20, 1)
-input = Variable(Tensor(100, 10).normal_())
-target = Variable(Tensor(100).normal_())
+input = Tensor(100, 10).normal_()
+target = Tensor(100, 1).normal_()
 output = mlp(input)
 criterion = nn.MSELoss()
 loss = criterion(output, target)
@@ -61,15 +60,18 @@ agtree2dot.save_dot(loss,
 print('Generated mlp.dot')
 
 try:
-    fontname='Computer Modern'
-    fontsize=12
+
+    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)