From ea9e3b0f976da1d5f7e16ce008ae58eb7db689fb Mon Sep 17 00:00:00 2001
From: Yuheng Qiu <115010216@link.cuhk.edu.cn>
Date: Wed, 10 Jun 2020 23:18:00 -0400
Subject: [PATCH] change scheduler

---
 datasets/continuumLS.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/datasets/continuumLS.py b/datasets/continuumLS.py
index d6203f8..48e9dc7 100644
--- a/datasets/continuumLS.py
+++ b/datasets/continuumLS.py
@@ -10,7 +10,7 @@
 from dgl.data import citegrh
 from itertools  import compress
 from torchvision.datasets import VisionDataset
-from sklearn.preprocessing import StandardScaler
+from sklearn.preprocessing import StandardScaler, MinMaxScaler
 
 
 class ContinuumLS(VisionDataset):
@@ -61,7 +61,7 @@ def load_data(self, prefix, normalize=True):
         assert len(class_map) == feats.shape[0]
         train_nodes = np.array(list(set(adj_train.nonzero()[0])))
         train_feats = feats[train_nodes]
-        scaler = StandardScaler()
+        scaler = MinMaxScaler()
         scaler.fit(train_feats)
         feats = scaler.transform(feats)
         return adj_full, adj_train, feats, class_map, role