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