clean code
This commit is contained in:
		| @@ -44,9 +44,6 @@ class BasePyTorchRegressor(BasePyTorchModel): | ||||
|             dk.data_dictionary["prediction_features"], | ||||
|             device=self.device | ||||
|         ) | ||||
|         logger.info(self.model.model) | ||||
|         logger.info(self.model.model) | ||||
|  | ||||
|         y = self.model.model(x) | ||||
|         pred_df = DataFrame(y.detach().numpy(), columns=[dk.label_list[0]]) | ||||
|         return (pred_df, dk.do_predict) | ||||
|   | ||||
		Reference in New Issue
	
	Block a user