Add Monotonic function toy example

matthewfeickert · Dec 15, 2019 · 7404356 · 7404356
1 parent 0aae261
commit 7404356
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 187 deletions.
diff --git a/lib/DrawContour.py b/lib/DrawContour.py
diff --git a/lib/QuadTree.py b/lib/QuadTree.py
diff --git a/models/UMNN/MonotonicNN.py b/models/UMNN/MonotonicNN.py
@@ -50,5 +50,5 @@ def forward(self, x, h):
         x0 = torch.zeros(x.shape).to(self.device)
         out = self.net(h)
         offset = out[:, [0]]
-        scaling = out[:, [1]]
+        scaling = torch.exp(out[:, [1]])
         return scaling*ParallelNeuralIntegral.apply(x0, x, self.integrand, _flatten(self.integrand.parameters()), h, self.nb_steps) + offset