Merge pull request #81 from eustomaqua/master

Correct the random_seed in ``classify.py''
thunlp · Aug 12, 2019 · d9cbf34 · d9cbf34
2 parents 1e5eae6 + e717872
commit d9cbf34
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/openne/__main__.py b/src/openne/__main__.py
@@ -172,7 +172,7 @@ def main(args):
         print("Training classifier using {:.2f}% nodes...".format(
             args.clf_ratio*100))
         clf = Classifier(vectors=vectors, clf=LogisticRegression())
-        clf.split_train_evaluate(X, Y, args.clf_ratio)
+        clf.split_train_evaluate(X, Y, args.clf_ratio, seed=0)
 
 
 if __name__ == "__main__":

diff --git a/src/openne/classify.py b/src/openne/classify.py
@@ -51,7 +51,7 @@ def predict(self, X, top_k_list):
         Y = self.clf.predict(X_, top_k_list=top_k_list)
         return Y
 
-    def split_train_evaluate(self, X, Y, train_precent, seed=0):
+    def split_train_evaluate(self, X, Y, train_precent, seed=None):
         state = numpy.random.get_state()
 
         training_size = int(train_precent * len(X))