diff options
author | Nicholas Leonard <nick@nikopia.org> | 2015-03-26 04:16:18 +0300 |
---|---|---|
committer | Nicholas Leonard <nick@nikopia.org> | 2015-03-26 04:23:09 +0300 |
commit | 8492652c78d712608082cadba803d9357b3ea9bf (patch) | |
tree | 70cb550cc92733b9ffa8405302f417340256d8b2 /CosineEmbeddingCriterion.lua | |
parent | 09168424f127407f05f69bc8a34a3deab91a209a (diff) |
Criterion doc++
Diffstat (limited to 'CosineEmbeddingCriterion.lua')
-rw-r--r-- | CosineEmbeddingCriterion.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/CosineEmbeddingCriterion.lua b/CosineEmbeddingCriterion.lua index 2dc3e9c..d5e0e70 100644 --- a/CosineEmbeddingCriterion.lua +++ b/CosineEmbeddingCriterion.lua @@ -2,7 +2,7 @@ local CosineEmbeddingCriterion, parent = torch.class('nn.CosineEmbeddingCriterio function CosineEmbeddingCriterion:__init(margin) parent.__init(self) - margin=margin or 0 + margin = margin or 0 self.margin = margin self.gradInput = {torch.Tensor(), torch.Tensor()} end @@ -15,7 +15,7 @@ function CosineEmbeddingCriterion:updateOutput(input,y) self.w32 = input2:dot(input2) self.w3 = math.sqrt(self.w32) self.output = self.w1/self.w2/self.w3 - if y==-1 then + if y == -1 then self.output = math.max(0, self.output - self.margin); else self.output = 1 - self.output |