3 #########################################################################
4 # This program is free software: you can redistribute it and/or modify #
5 # it under the terms of the version 3 of the GNU General Public License #
6 # as published by the Free Software Foundation. #
8 # This program is distributed in the hope that it will be useful, but #
9 # WITHOUT ANY WARRANTY; without even the implied warranty of #
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU #
11 # General Public License for more details. #
13 # You should have received a copy of the GNU General Public License #
14 # along with this program. If not, see <http://www.gnu.org/licenses/>. #
16 # Written by and Copyright (C) Francois Fleuret #
17 # Contact <francois.fleuret@idiap.ch> for comments & bug reports #
18 #########################################################################
23 from torch.nn import functional as fn
24 from torch import Tensor
25 from torch.autograd import Variable
26 from torch.nn import Module
31 def __init__(self, input_dim, hidden_dim, output_dim):
32 super(MLP, self).__init__()
33 self.fc1 = nn.Linear(input_dim, hidden_dim)
34 self.fc2 = nn.Linear(hidden_dim, output_dim)
43 input = Variable(Tensor(100, 10).normal_())
44 target = Variable(Tensor(100).normal_())
46 criterion = nn.MSELoss()
47 loss = criterion(output, target)
49 agtree2dot.save_dot(loss,
54 mlp.fc1.weight: 'weight1',
55 mlp.fc1.bias: 'bias1',
56 mlp.fc2.weight: 'weight2',
57 mlp.fc2.bias: 'bias2',
59 open('./mlp.dot', 'w'))
61 print('Generated mlp.dot')
65 fontname = 'Computer Modern'
67 subprocess.check_call(['dot', 'mlp.dot',
70 '-Efontname=' + fontname, '-Efontsize=' + str(fontsize),
71 '-Nfontname=' + fontname, '-Nfontsize=' + str(fontsize),
74 except subprocess.CalledProcessError:
76 print('Calling the dot command failed. Is Graphviz installed?')
79 print('Generated mlp.pdf')