clean code
This commit is contained in:
parent
c137666230
commit
36a0a14a23
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user