201df9a84a2585537e0b7f1db4b97244bc7a4e2d,autokeras/hypermodels/heads.py,ClassificationHead,__init__,#ClassificationHead#Any#Any#Any#Any#Any#,39

Before Change

        if not self.metrics:
            self.metrics = ["accuracy"]
        self.dropout_rate = dropout_rate
        self.set_loss()

    def set_loss(self):
        if not self.num_classes:
            return

After Change

        self.dropout_rate = dropout_rate
        if metrics is None:
            metrics = ["accuracy"]
        if loss is None:
            loss = self.infer_loss()
        super().__init__(loss=loss,
                         metrics=metrics,
                         **kwargs)

Instances