Artificial Intelligence, Machine Learning, Deep Learning, Image Classification, Heartbeat.
num_classes=num_classes) self.f1_score = F1Score(task="binary", num_classes=num_classes) self.history = {'train_loss':[],'train_acc':[],'val_loss':[],'val_acc' : []} # Defining the model architecture self.pretrain_model = resnet50(weights=ResNet50_Weights.DEFAULT) self.pretrain_model.eval() for