Unverified Commit 79ceb479 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #2671 from arunppsg/model_print

fixes attribute error from sklearn in print model
parents 49e6ee64 abe2d888
......@@ -175,6 +175,9 @@ class KerasModel(Model):
the Weights & Biases logger object used to log data and metrics
"""
super(KerasModel, self).__init__(model=model, model_dir=model_dir, **kwargs)
self.loss = loss # not used
self.learning_rate = learning_rate # not used
self.output_types = output_types # not used
if isinstance(loss, Loss):
self._loss_fn: LossFn = _StandardLoss(model, loss)
else:
......
......@@ -175,6 +175,9 @@ class TorchModel(Model):
the Weights & Biases logger object used to log data and metrics
"""
super(TorchModel, self).__init__(model=model, model_dir=model_dir, **kwargs)
self.loss = loss # not used
self.learning_rate = learning_rate # not used
self.output_types = output_types # not used
if isinstance(loss, Loss):
self._loss_fn: LossFn = _StandardLoss(self, loss)
else:
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment