From 3d72314f955b8c5f1bd2f10276bd95dfaeda812c Mon Sep 17 00:00:00 2001
From: Ayushman Pani <126395065+ayushmanpani@users.noreply.github.com>
Date: Sun, 29 Dec 2024 00:54:09 +0530
Subject: [PATCH] CNN_ECG.py

resolved the error:
Arguments target and output must have the same rank (ndim). Received: target.shape=(32, 1, 4), output.shape=(32, 4)
---
 CNN_ECG.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/CNN_ECG.py b/CNN_ECG.py
index 89e6a68..b64240b 100644
--- a/CNN_ECG.py
+++ b/CNN_ECG.py
@@ -53,7 +53,7 @@ def change(x):
         dummy = np.zeros((number_of_classes))
         dummy[int(target_train[i])] = 1
         Label_set[i, :] = dummy'''
-    Label_set = np.eye(number_of_classes)[target_train.astype(int)]
+    Label_set = np.eye(number_of_classes)[target_train.astype(int)].reshape(-1, number_of_classes)
         
 elif sys.argv[1] == 'mit':
     print('In proces...')