From 906a3aa1a1e95ea80c6fcdee4246f904216cc699 Mon Sep 17 00:00:00 2001
From: Chengxi Zhang <v-zhanch@microsoft.com>
Date: Sun, 10 Nov 2019 13:02:00 +0800
Subject: [PATCH] Freeze model performance by applying the same one-hot
 transform strategy every time.

---
 pygcn/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pygcn/utils.py b/pygcn/utils.py
index 9b53c5b..527ad2d 100644
--- a/pygcn/utils.py
+++ b/pygcn/utils.py
@@ -4,7 +4,7 @@
 
 
 def encode_onehot(labels):
-    classes = set(labels)
+    classes = 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)),