Update. master
authorFrancois Fleuret <francois@fleuret.org>
Sun, 20 May 2018 21:02:26 +0000 (23:02 +0200)
committerFrancois Fleuret <francois@fleuret.org>
Sun, 20 May 2018 21:02:26 +0000 (23:02 +0200)
README.md
mlp.pdf

index f290eca..6980672 100644 (file)
--- a/README.md
+++ b/README.md
@@ -20,12 +20,9 @@ resulting graph.
 A typical use is provided in [mlp.py](https://fleuret.org/git-extract/agtree2dot/mlp.py):
 
 ```python
-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
@@ -43,8 +40,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)
@@ -64,7 +61,8 @@ agtree2dot.save_dot(loss,
 print('Generated mlp.dot')
 
 try:
-    subprocess.check_call(["dot", "mlp.dot", "-Lg", "-T", "pdf", "-o", "mlp.pdf" ])
+    subprocess.check_call(['dot', 'mlp.dot', '-Lg', '-T', 'pdf', '-o', 'mlp.pdf' ])
+
 except subprocess.CalledProcessError:
     print('Calling the dot command failed. Is Graphviz installed?')
     sys.exit(1)
diff --git a/mlp.pdf b/mlp.pdf
index a72a3b7..0a7b858 100644 (file)
Binary files a/mlp.pdf and b/mlp.pdf differ