[Bugfix] removing non-determinism in data loading (dmlc#413)

[Bugfix] Removing non-determinism in data loading
liuweiping2020 · Feb 26, 2019 · c07ae34 · c07ae34
1 parent 565f0c8
commit c07ae34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/dgl/data/citation_graph.py b/python/dgl/data/citation_graph.py
@@ -389,7 +389,7 @@ def _normalize(mx):
     return mx
 
 def _encode_onehot(labels):
-    classes = set(labels)
+    classes = list(sorted(set(labels)))
     classes_dict = {c: np.identity(len(classes))[i, :] for i, c in
                     enumerate(classes)}
     labels_onehot = np.array(list(map(classes_dict.get, labels)),