Skip to content

Commit af28292

Browse files
Sayed-Hossein-HosseiniSayed-Hossein-Hosseini
committed
Load Datasets
1 parent f1097aa commit af28292

1 file changed

Lines changed: 146 additions & 29 deletions

File tree

‎RhythmAttention_Hybrid_CNN_Transformer_Architecture_for_Arrhythmia_Classification.ipynb‎

Lines changed: 146 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -1,55 +1,172 @@
11
{
2-
"nbformat": 4,
3-
"nbformat_minor": 0,
4-
"metadata": {
5-
"colab": {
6-
"provenance": [],
7-
"authorship_tag": "ABX9TyOVKJ8hw3MO9nMPohhu6IFS",
8-
"include_colab_link": true
9-
},
10-
"kernelspec": {
11-
"name": "python3",
12-
"display_name": "Python 3"
13-
},
14-
"language_info": {
15-
"name": "python"
16-
}
17-
},
182
"cells": [
193
{
204
"cell_type": "markdown",
215
"metadata": {
22-
"id": "view-in-github",
23-
"colab_type": "text"
6+
"colab_type": "text",
7+
"id": "view-in-github"
248
},
259
"source": [
2610
"<a href=\"https://colab.research.google.com/github/Sayed-Hossein-Hosseini/RhythmAttention_Hybrid_CNN-Transformer_Architecture_for_Arrhythmia_Classification/blob/master/RhythmAttention_Hybrid_CNN_Transformer_Architecture_for_Arrhythmia_Classification.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
2711
]
2812
},
2913
{
3014
"cell_type": "markdown",
31-
"source": [
32-
"# **RhythmAttention: Hybrid CNN-Transformer Architecture for Arrhythmia Classification**"
33-
],
3415
"metadata": {
3516
"id": "OF_FhXOiVvlU"
36-
}
17+
},
18+
"source": [
19+
"# **RhythmAttention: Hybrid CNN-Transformer Architecture for Arrhythmia Classification**"
20+
]
3721
},
3822
{
3923
"cell_type": "markdown",
40-
"source": [],
4124
"metadata": {
4225
"id": "tgCNPtKDWJWn"
43-
}
26+
},
27+
"source": [
28+
"## **Libraries**"
29+
]
4430
},
4531
{
4632
"cell_type": "code",
47-
"execution_count": null,
33+
"execution_count": 1,
4834
"metadata": {
4935
"id": "fknaaWAzVqXF"
5036
},
51-
"outputs": [],
52-
"source": []
37+
"outputs": [
38+
{
39+
"name": "stdout",
40+
"output_type": "stream",
41+
"text": [
42+
"Requirement already satisfied: tensorflow in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (2.20.0)\n",
43+
"Requirement already satisfied: absl-py>=1.0.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (2.4.0)\n",
44+
"Requirement already satisfied: astunparse>=1.6.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (1.6.3)\n",
45+
"Requirement already satisfied: flatbuffers>=24.3.25 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (25.12.19)\n",
46+
"Requirement already satisfied: gast!=0.5.0,!=0.5.1,!=0.5.2,>=0.2.1 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (0.7.0)\n",
47+
"Requirement already satisfied: google_pasta>=0.1.1 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (0.2.0)\n",
48+
"Requirement already satisfied: libclang>=13.0.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (18.1.1)\n",
49+
"Requirement already satisfied: opt_einsum>=2.3.2 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (3.4.0)\n",
50+
"Requirement already satisfied: packaging in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (25.0)\n",
51+
"Requirement already satisfied: protobuf>=5.28.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (6.33.5)\n",
52+
"Requirement already satisfied: requests<3,>=2.21.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (2.32.5)\n",
53+
"Requirement already satisfied: setuptools in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (80.10.2)\n",
54+
"Requirement already satisfied: six>=1.12.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (1.17.0)\n",
55+
"Requirement already satisfied: termcolor>=1.1.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (3.3.0)\n",
56+
"Requirement already satisfied: typing_extensions>=3.6.6 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (4.15.0)\n",
57+
"Requirement already satisfied: wrapt>=1.11.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (2.1.1)\n",
58+
"Requirement already satisfied: grpcio<2.0,>=1.24.3 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (1.76.0)\n",
59+
"Requirement already satisfied: tensorboard~=2.20.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (2.20.0)\n",
60+
"Requirement already satisfied: keras>=3.10.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (3.13.2)\n",
61+
"Requirement already satisfied: numpy>=1.26.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (1.26.4)\n",
62+
"Requirement already satisfied: h5py>=3.11.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (3.15.1)\n",
63+
"Requirement already satisfied: ml_dtypes<1.0.0,>=0.5.1 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorflow) (0.5.4)\n",
64+
"Requirement already satisfied: charset_normalizer<4,>=2 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from requests<3,>=2.21.0->tensorflow) (3.4.4)\n",
65+
"Requirement already satisfied: idna<4,>=2.5 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from requests<3,>=2.21.0->tensorflow) (3.11)\n",
66+
"Requirement already satisfied: urllib3<3,>=1.21.1 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from requests<3,>=2.21.0->tensorflow) (2.5.0)\n",
67+
"Requirement already satisfied: certifi>=2017.4.17 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from requests<3,>=2.21.0->tensorflow) (2026.1.4)\n",
68+
"Requirement already satisfied: markdown>=2.6.8 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorboard~=2.20.0->tensorflow) (3.10.1)\n",
69+
"Requirement already satisfied: pillow in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorboard~=2.20.0->tensorflow) (12.1.0)\n",
70+
"Requirement already satisfied: tensorboard-data-server<0.8.0,>=0.7.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorboard~=2.20.0->tensorflow) (0.7.2)\n",
71+
"Requirement already satisfied: werkzeug>=1.0.1 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from tensorboard~=2.20.0->tensorflow) (3.1.5)\n",
72+
"Requirement already satisfied: wheel<1.0,>=0.23.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from astunparse>=1.6.0->tensorflow) (0.46.3)\n",
73+
"Requirement already satisfied: rich in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from keras>=3.10.0->tensorflow) (14.3.2)\n",
74+
"Requirement already satisfied: namex in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from keras>=3.10.0->tensorflow) (0.1.0)\n",
75+
"Requirement already satisfied: optree in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from keras>=3.10.0->tensorflow) (0.18.0)\n",
76+
"Requirement already satisfied: markupsafe>=2.1.1 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from werkzeug>=1.0.1->tensorboard~=2.20.0->tensorflow) (3.0.3)\n",
77+
"Requirement already satisfied: markdown-it-py>=2.2.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from rich->keras>=3.10.0->tensorflow) (4.0.0)\n",
78+
"Requirement already satisfied: pygments<3.0.0,>=2.13.0 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from rich->keras>=3.10.0->tensorflow) (2.19.2)\n",
79+
"Requirement already satisfied: mdurl~=0.1 in /home/zeus/miniconda3/envs/cloudspace/lib/python3.12/site-packages (from markdown-it-py>=2.2.0->rich->keras>=3.10.0->tensorflow) (0.1.2)\n",
80+
"\n",
81+
"\u001b[1m[\u001b[0m\u001b[34;49mnotice\u001b[0m\u001b[1;39;49m]\u001b[0m\u001b[39;49m A new release of pip is available: \u001b[0m\u001b[31;49m26.0\u001b[0m\u001b[39;49m -> \u001b[0m\u001b[32;49m26.0.1\u001b[0m\n",
82+
"\u001b[1m[\u001b[0m\u001b[34;49mnotice\u001b[0m\u001b[1;39;49m]\u001b[0m\u001b[39;49m To update, run: \u001b[0m\u001b[32;49mpip install --upgrade pip\u001b[0m\n"
83+
]
84+
}
85+
],
86+
"source": [
87+
"!pip install tensorflow"
88+
]
89+
},
90+
{
91+
"cell_type": "code",
92+
"execution_count": 2,
93+
"metadata": {},
94+
"outputs": [
95+
{
96+
"name": "stderr",
97+
"output_type": "stream",
98+
"text": [
99+
"2026-02-08 17:57:33.534489: I tensorflow/core/util/port.cc:153] oneDNN custom operations are on. You may see slightly different numerical results due to floating-point round-off errors from different computation orders. To turn them off, set the environment variable `TF_ENABLE_ONEDNN_OPTS=0`.\n",
100+
"2026-02-08 17:57:33.583704: I tensorflow/core/platform/cpu_feature_guard.cc:210] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.\n",
101+
"To enable the following instructions: AVX2 AVX512F AVX512_VNNI FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.\n",
102+
"2026-02-08 17:57:34.748647: I tensorflow/core/util/port.cc:153] oneDNN custom operations are on. You may see slightly different numerical results due to floating-point round-off errors from different computation orders. To turn them off, set the environment variable `TF_ENABLE_ONEDNN_OPTS=0`.\n"
103+
]
104+
}
105+
],
106+
"source": [
107+
"import pandas as pd\n",
108+
"import numpy as np\n",
109+
"import tensorflow as tf\n",
110+
"from sklearn.utils import class_weight"
111+
]
112+
},
113+
{
114+
"cell_type": "markdown",
115+
"metadata": {},
116+
"source": [
117+
"## **Datasets**"
118+
]
119+
},
120+
{
121+
"cell_type": "code",
122+
"execution_count": 3,
123+
"metadata": {},
124+
"outputs": [
125+
{
126+
"name": "stdout",
127+
"output_type": "stream",
128+
"text": [
129+
"X_train shape: (87554, 187, 1)\n",
130+
"X_test shape: (21892, 187, 1)\n"
131+
]
132+
}
133+
],
134+
"source": [
135+
"# 1. Load the dataset (Ensure files are in the working directory)\n",
136+
"train_df = pd.read_csv('Datasets/mitbih_train.csv', header=None)\n",
137+
"test_df = pd.read_csv('Datasets/mitbih_test.csv', header=None)\n",
138+
"\n",
139+
"# 2. Separate features (ECG signal) and labels\n",
140+
"# The last column (index 187) is the class label (0 to 4)\n",
141+
"X_train = train_df.iloc[:, :187].values\n",
142+
"y_train = train_df.iloc[:, 187].astype(int).values\n",
143+
"\n",
144+
"X_test = test_df.iloc[:, :187].values\n",
145+
"y_test = test_df.iloc[:, 187].astype(int).values\n",
146+
"\n",
147+
"# 3. Reshape signals to 3D for Conv1D layers: (Samples, TimeSteps, Channels)\n",
148+
"X_train = X_train.reshape(X_train.shape[0], X_train.shape[1], 1)\n",
149+
"X_test = X_test.reshape(X_test.shape[0], X_test.shape[1], 1)\n",
150+
"\n",
151+
"print(f\"X_train shape: {X_train.shape}\") # Expected: (87554, 187, 1)\n",
152+
"print(f\"X_test shape: {X_test.shape}\") # Expected: (21892, 187, 1)"
153+
]
53154
}
54-
]
55-
}
155+
],
156+
"metadata": {
157+
"colab": {
158+
"authorship_tag": "ABX9TyOVKJ8hw3MO9nMPohhu6IFS",
159+
"include_colab_link": true,
160+
"provenance": []
161+
},
162+
"kernelspec": {
163+
"display_name": "Python 3",
164+
"name": "python3"
165+
},
166+
"language_info": {
167+
"name": "python"
168+
}
169+
},
170+
"nbformat": 4,
171+
"nbformat_minor": 0
172+
}

0 commit comments

Comments
 (0)