Outdated pandas api, change as_matrix() to .values

ignacio-rocco · Feb 28, 2019 · c9cf842 · c9cf842
1 parent 0bd02d4
commit c9cf842
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/im_pair_dataset.py b/lib/im_pair_dataset.py
@@ -32,8 +32,8 @@ def __init__(self, dataset_csv_path, dataset_csv_file, dataset_image_path, datas
  self.train_data = self.train_data.iloc[0:dataset_size,:]
  self.img_A_names = self.train_data.iloc[:,0]
  self.img_B_names = self.train_data.iloc[:,1] 
- self.set = self.train_data.iloc[:,2].as_matrix()
- self.flip = self.train_data.iloc[:, 3].as_matrix().astype('int')
+ self.set = self.train_data.iloc[:,2].values
+ self.flip = self.train_data.iloc[:, 3].values.astype('int')
  self.dataset_image_path = dataset_image_path 
  self.transform = transform
  # no cuda as dataset is called from CPU threads in dataloader and produces confilct