Просмотр исходного кода

Removed not used Neighbourhood search in _BMB function.

Kristian Schultz 4 лет назад
Родитель
Сommit
af90286eed
1 измененных файлов с 2 добавлено и 2 удалено
  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))