fix predict bug

kexinhuang12345 · Oct 26, 2021 · 1dcef58 · 1dcef58
1 parent 27283db
commit 1dcef58
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/DeepPurpose/DDI.py b/DeepPurpose/DDI.py
@@ -368,7 +368,7 @@ def predict(self, df_data):
 			pd.DataFrame
 		'''
 		print('predicting...')
-		info = data_process_loader(df_data.index.values, df_data.Label.values, df_data, **self.config)
+		info = data_process_DDI_loader(df_data.index.values, df_data.Label.values, df_data, **self.config)
 		self.model.to(device)
 		params = {'batch_size': self.config['batch_size'],
 				'shuffle': False,

diff --git a/DeepPurpose/PPI.py b/DeepPurpose/PPI.py
@@ -353,14 +353,13 @@ def predict(self, df_data):
 			pd.DataFrame
 		'''
 		print('predicting...')
-		info = data_process_loader(df_data.index.values, df_data.Label.values, df_data, **self.config)
 		self.model.to(device)
+		info = data_process_PPI_loader(df_data.index.values, df_data.Label.values, df_data, **self.config)
 		params = {'batch_size': self.config['batch_size'],
 				'shuffle': False,
 				'num_workers': self.config['num_workers'],
 				'drop_last': False,
 				'sampler':SequentialSampler(info)}
-
 		generator = data.DataLoader(info, **params)
 
 		score = self.test_(generator, self.model, repurposing_mode = True)

diff --git a/DeepPurpose/ProteinPred.py b/DeepPurpose/ProteinPred.py
@@ -342,7 +342,7 @@ def predict(self, df_data, verbose = True):
 		'''
 		if verbose:
 			print('predicting...')
-		info = data_process_loader_Property_Prediction(df_data.index.values, df_data.Label.values, df_data, **self.config)
+		info = data_process_loader_Protein_Prediction(df_data.index.values, df_data.Label.values, df_data, **self.config)
 		self.model.to(device)
 		params = {'batch_size': self.config['batch_size'],
 				'shuffle': False,