clean code

This commit is contained in:
Yinon Polak 2023-04-03 16:26:42 +03:00
parent c137666230
commit 36a0a14a23

View File

@ -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)