Update featurization.py

swarajpande5 · Apr 12, 2024 · 8dbd34f · 8dbd34f
1 parent 07fafa0
commit 8dbd34f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/code/featurization.py b/code/featurization.py
@@ -16,7 +16,7 @@ def featurization():
 
     # Create PCA object of the 20 most important components
     print("Creating PCA object...")
-    pca = PCA(n_components=20, whiten=True)
+    pca = PCA(n_components=25, whiten=True)
     pca.fit(train_data[:, 1:])
     train_labels = train_data[:, 0].reshape([train_data.shape[0], 1])
     test_labels = test_data[:, 0].reshape([test_data.shape[0], 1])
@@ -35,4 +35,4 @@ def featurization():
 
 
 if __name__ == '__main__':
-    featurization()
+    featurization()