weiyuqing/ECG_by_Diffenert_CNN_Test
1
0 Fork 0
mirror of https://github.com/This-weiyuqing/ECG_by_Diffenert_CNN_Test.git synced 2026-10-18 03:13:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update readme

Browse Source
This commit is contained in:
weiyuqing committed 2023-05-16 11:17:34 +08:00
1 parent e44ea0019c
commit 05542c2ad9
2 files changed
+35 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

some-test/test_coplit.py
+27
View File
@@ -0,0 +1,27 @@
def get_MAX():
MAX = 0
for i in range(0, len(X_train)):
if MAX < len(X_train[i]):
MAX = len(X_train[i])
for i in range(0, len(X_test)):
if MAX < len(X_test[i]):
MAX = len(X_test[i])
return MAX
def made_CNN():
# 1D CNN neural network
model = Sequential()
model.add(Conv1D(32, 5, activation='relu', input_shape=(MAX, 1)))
model.add(MaxPooling1D(2))
model.add(Conv1D(64, 5, activation='relu'))
model.add(MaxPooling1D(2))
model.add(Conv1D(128, 5, activation='relu'))
model.add(MaxPooling1D(2))
model.add(Conv1D(256, 5, activation='relu'))
model.add(MaxPooling1D(2))
model.add(Conv1D(512, 5, activation='relu'))
model.add(MaxPooling1D(2))
model.add(Flatten())
model.add(Dense(512, activation='relu'))
model.add(Dense(5, activation='softmax'))
model.summary()
return model