Fixed wrt the source.
authorFrancois Fleuret <francois@fleuret.org>
Wed, 8 Mar 2017 10:47:16 +0000 (11:47 +0100)
committerFrancois Fleuret <francois@fleuret.org>
Wed, 8 Mar 2017 10:47:16 +0000 (11:47 +0100)
README.md

index 4c219b7..9b4017c 100644 (file)
--- a/README.md
+++ b/README.md
@@ -46,7 +46,7 @@ criterion = nn.MSELoss()
 loss = criterion(output, target)
 
 agtree2dot.save_dot(loss,
-                    { input: 'input', loss: 'loss' },
+                    { input: 'input', target: 'target', loss: 'loss' },
                     open('./mlp.dot', 'w'))
 ```