Fix some small bug in TransH.

AllenWangle · Aug 15, 2015 · 160ae3a · 160ae3a
1 parent 954e6e4
commit 160ae3a
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/PTransE/.DS_Store b/PTransE/.DS_Store
diff --git a/TransH/Train_TransH.cpp b/TransH/Train_TransH.cpp
@@ -286,8 +286,8 @@ class Train{
             relation_tmp[rel][ii]-=belta*rate*x;
             entity_tmp[e1][ii]-=belta*rate*x;
             entity_tmp[e2][ii]+=belta*rate*x;
-            A[rel][ii]+=belta*rate*x*tmp1;
-            A[rel][ii]-=belta*rate*x*tmp2;
+            A_tmp[rel][ii]+=belta*rate*x*tmp1;
+            A_tmp[rel][ii]-=belta*rate*x*tmp2;
         }
         for (int ii=0; ii<n; ii++)
         {