Change scaler

CihanSoylu · Mar 18, 2020 · 9e5093a · 9e5093a
1 parent ba79113
commit 9e5093a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/project_utils/project_2_utils.py b/project_utils/project_2_utils.py
@@ -22,11 +22,13 @@ def load_data(n_features = 2):
 
     x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.3, stratify = y, random_state=2)
 
-    scaler = MinMaxScaler()
+    scaler = StandardScaler()
     scaler.fit(x_train)
     x_train = scaler.transform(x_train)
     x_test = scaler.transform(x_test)
 
+
+
     return x_train[:,:n_features], x_test[:,:n_features], y_train, y_test
 
 def plot_data(x,y):