From: Francois Fleuret <francois@fleuret.org>
Date: Sun, 20 May 2018 21:02:26 +0000 (+0200)
Subject: Update.
X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=agtree2dot.git;a=commitdiff_plain;h=765afb07c2781c92805461eb937d611fe03ca9bf

Update.
---

diff --git a/README.md b/README.md
index f290eca..6980672 100644
--- 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
Binary files a/mlp.pdf and b/mlp.pdf differ