Artificial Intelligence, Machine Learning, Deep Learning, Image Classification, Heartbeat.
CancerImageClassifier(pl.LightningModule): def __init__(self, learning_rate = 0.001, num_classes = 2): super().__init__() self.learning_rate = learning_rate self.loss_fn = nn.CrossEntropyLoss() self.num_classes = num_classes # Defining metrics self.accuracy = Accuracy(task="binary",