Sfoglia il codice sorgente

Removed not used Neighbourhood search in _BMB function.

Kristian Schultz 4 anni fa
parent
commit
af90286eed
1 ha cambiato i file con 2 aggiunte e 2 eliminazioni
  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))