update

kimmo1019 · Jan 20, 2021 · c1d0115 · c1d0115
1 parent 6d8da1f
commit c1d0115
Showing 1 changed file with 0 additions and 10 deletions.
diff --git a/main_clustering.py b/main_clustering.py
@@ -254,16 +254,6 @@ def evaluate(self,timestamp,batch_idx):
                 np.savez('{}/data_at_{}.npz'.format(self.save_dir, batch_idx+1),data_x_,data_x_onehot_,label_y)
             else:
                 np.savez('results/{}/data_pre.npz'.format(self.data),data_x_,data_x_onehot_,label_y)    
-                #generation
-                size = 186
-                bx, _ = self.x_sampler.train(size)
-                bx_onehot = np.zeros((size, self.nb_classes))
-                bx_onehot[:,5] = 1
-                y_pred = self.predict_y(bx,bx_onehot)
-                print('pred shape',y_pred.shape)
-                embeds,_ = self.predict_x(y_pred)
-                np.savez('results/{}/data_impute_{}.npz'.format(self.data,size),embeds,y_pred)    
-
         else:
             if is_train:
                 np.savez('{}/data_at_{}.npz'.format(self.save_dir, batch_idx+1),data_x_,data_x_onehot_)