Add files via upload

Author: JingweiToo

Demo_PSO.py

@@ -38,13 +38,8 @@
 mdl.fit(x_train, y_train)
 
 # accuracy
-pred      = mdl.predict(x_valid)
-correct   = 0
-for i in range(num_valid):
-    if pred[i] == y_valid[i]:
-        correct += 1
-
-Acc = correct / num_valid
+y_pred    = mdl.predict(x_valid)
+Acc       = np.sum(y_valid == y_pred)  / num_valid
 print("Accuracy:", 100 * Acc)
 
 # number of selected features

FS/functionHO.py

@@ -24,14 +24,9 @@ def error_rate(xtrain, ytrain, x, opts):
     mdl     = KNeighborsClassifier(n_neighbors = k)
     mdl.fit(xtrain, ytrain)
     # Prediction
-    pred    = mdl.predict(xvalid)
-    correct = 0
-    for i in range(num_valid):
-        if pred[i] == yvalid[i]:
-            correct += 1
-    
-    accuracy = correct / num_valid
-    error    = 1 - accuracy
+    ypred   = mdl.predict(xvalid)
+    acc     = np.sum(yvalid == ypred) / num_valid
+    error   = 1 - acc
 
     return error
 

FS/ga.py

@@ -138,4 +138,4 @@ def jfs(xtrain, ytrain, opts):
 
 
 
-    
+