Przeglądaj źródła

Removed not used Neighbourhood search in _BMB function.

Kristian Schultz 4 lat temu
rodzic
commit
af90286eed
1 zmienionych plików z 2 dodań i 2 usunięć
  1. 2 2
      library/convGAN2.py

+ 2 - 2
library/convGAN2.py

@@ -345,8 +345,8 @@ class ConvGAN2(GanBaseClass):
         ## neb -> oversampling neighbourhood
         ## gen -> convex combinations generated from each neighbourhood
 
-        neigh = NearestNeighbors(self.neb)
-        neigh.fit(data_maj)
+        #neigh = NearestNeighbors(self.neb)
+        #neigh.fit(data_maj)
         # bmbi = [
         #     neigh.kneighbors([data_min[i]], self.neb, return_distance=False)
         #     for i in range(len(data_min))