Parallelized Link Prediction

luigiba · Aug 5, 2019 · d0029ba · d0029ba
1 parent cdd9cd6
commit d0029ba
Show file tree

Hide file tree

Showing 22 changed files with 1,152 additions and 1,165 deletions.
diff --git a/.idea/misc.xml b/.idea/misc.xml
diff --git a/.idea/modules.xml b/.idea/modules.xml
diff --git a/.idea/vcs.xml b/.idea/vcs.xml
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/Config.py b/Config.py
diff --git a/OpenKE_new_Spark.iml → OpenKEonSpark.iml b/OpenKE_new_Spark.iml → OpenKEonSpark.iml
@@ -1,5 +1,5 @@
 <?xml version="1.0" encoding="UTF-8"?>
-<module type="WEB_MODULE" version="4">
+<module type="PYTHON_MODULE" version="4">
   <component name="NewModuleRootManager" inherit-compiler-output="true">
     <exclude-output />
     <content url="file://$MODULE_DIR$" />

diff --git a/__pycache__/Config.cpython-36.pyc b/__pycache__/Config.cpython-36.pyc
diff --git a/__pycache__/distribute_training.cpython-36.pyc b/__pycache__/distribute_training.cpython-36.pyc
diff --git a/base/Base.cpp b/base/Base.cpp
@@ -48,10 +48,6 @@ INT getTrainTotal_();
 extern "C"
 INT getBatchTotal();
 
-//EDIT
-extern "C"
-void setConstrainFlag(INT con);
-
 extern "C"
 void randReset();
 
@@ -130,7 +126,6 @@ void* getBatch(void* con) {
 
 		}
 
-
 		for (INT times = 0; times < negRelRate; times++) {
 			batch_h[batch + last] = trainList_no[i].h;
 			batch_t[batch + last] = trainList_no[i].t;

diff --git a/base/Setting.h b/base/Setting.h
@@ -108,20 +108,11 @@ INT getValidTotal() {
 */
 
 INT bernFlag = 0;
-//EDIT
-INT constrainFlag = 0;
 
 extern "C"
 void setBern(INT con) {
 	bernFlag = con;
 }
 
-//EDIT
-extern "C"
-void setConstrainFlag(INT con) {
-	constrainFlag = con;
-}
-
-
 
 #endif