簡介一套用Python實(shí)現(xiàn)的心電算法工程面向生物醫(yī)學(xué)工程學(xué)習(xí)者、算法入門者與醫(yī)療數(shù)據(jù)分析人員解決心電信號去噪、R波定位、心率計(jì)算及心律失常識別等核心問題。代碼涵蓋巴特沃茲與卡爾曼濾波器、小波變換R波檢測、R-R間期心率計(jì)算并集成SVM、隨機(jī)森林和神經(jīng)網(wǎng)絡(luò)分類器可完成房顫、室顫室速等病理狀態(tài)識別與偽差干擾研究。包內(nèi)共109個(gè)文件以py源碼、dat/hea/atr心電數(shù)據(jù)文件為主要組成輔以pdf說明文檔、可視化圖表和測試用例壓縮包大小34.06MB。已有96人學(xué)習(xí)下載適合結(jié)合MIT-BIH數(shù)據(jù)庫驗(yàn)證完整算法流程也可作為課程項(xiàng)目或畢業(yè)設(shè)計(jì)的參考實(shí)現(xiàn)。1. 這份 Python 心電資源包里裝了什么從濾波到房顫識別的完整鏈路看到資源列表里那串200.atr、203.atr、105.atr老讀者應(yīng)該直接反應(yīng)過來了——這是 MIT-BIH 心律失常數(shù)據(jù)庫的注釋文件。我拆過不少 Python 心電算法開源工程這套資源的價(jià)值不在于某個(gè)單點(diǎn)算法多炫而在于它把濾波、R 波檢測、心率計(jì)算、特征提取、心律失常分類到房顫/室顫識別這條鏈路完整打通了。適合三類人剛接觸生物醫(yī)學(xué)信號處理的學(xué)生、要做心電預(yù)處理和搏動(dòng)檢測的算法工程師、以及想拿公開數(shù)據(jù)驗(yàn)證自己分類模型的從業(yè)者。它能讓你少走至少兩周彎路因?yàn)槊恳画h(huán)都有可運(yùn)行代碼和對應(yīng)的驗(yàn)證文件。2. 濾波不是玄學(xué)給 ECG 信號做零相位帶通與去基線漂移2.1 為什么不用滑動(dòng)平均而是巴特沃斯 filtfilt心電信號的主要噪聲來源就三類基線漂移頻率通常低于 0.5Hz、工頻干擾50Hz國內(nèi)電網(wǎng)、肌電偽差高頻能量集中在 100Hz 以上。很多人一上來就套滑動(dòng)窗口濾波那東西做平滑可以但會(huì)直接把 QRS 波群邊緣抹掉后面 R 波檢測的定位精度會(huì)肉眼可見地變差。心電處理里最穩(wěn)妥的方案是巴特沃斯帶通濾波器再用filtfilt做零相位濾波。filtfilt和普通lfilter的關(guān)鍵差別在于lfilter會(huì)引入與濾波器階數(shù)相關(guān)的群延遲相當(dāng)于把整段信號在時(shí)間軸上平移了一段而filtfilt先正向過一遍再反向過一遍相位畸變互相抵消R 波的位置不會(huì)因?yàn)闉V波而產(chǎn)生系統(tǒng)性偏移。在心率變異性分析這種對時(shí)間精度敏感的環(huán)節(jié)這是硬性要求。import numpy as np from scipy.signal import butter, filtfilt, iirnotch def bandpass_ecg(ecg, fs360.0, lowcut0.5, highcut50.0, order4): 零相位帶通濾波器默認(rèn)按 MIT-BIH 的 360Hz 采樣率設(shè)置 nyq fs / 2.0 b, a butter(order, [lowcut / nyq, highcut / nyq], btypeband) return filtfilt(b, a, ecg) def notch_50hz(ecg, fs360.0, f050.0, q30): 50Hz 工頻陷波q 值越大陷波帶寬越窄 b, a iirnotch(f0, q, fs) return filtfilt(b, a, ecg)帶通范圍選 0.5–50Hz 的依據(jù)0.5Hz 以上的范圍能保留大部分 ST 段和 T 波形態(tài)同時(shí)壓掉呼吸引起的基線漂移50Hz 以上的信號里 QRS 高頻分量已經(jīng)很少切掉不影響 R 波定位反而能削弱肌電噪聲。如果你的數(shù)據(jù)是 250Hz 采樣把highcut改到 40Hz 更穩(wěn)妥因?yàn)?50Hz 已經(jīng)在奈奎斯特頻率附近濾波器邊緣會(huì)變得很陡容易出現(xiàn)振鈴。2.2 去除基線漂移的另一種思路中值濾波與移動(dòng)平均巴特沃斯高通只能做全局去漂移如果碰到電極接觸不良導(dǎo)致的大幅低頻擺動(dòng)帶通濾波之后可能依然殘留一段弧線。常見的補(bǔ)充做法是用中值濾波估計(jì)基線再從原信號里減掉。中值濾波的優(yōu)勢是它不假設(shè)基線符合某個(gè)固定頻率范圍對突變型漂移更魯棒。from scipy.ndimage import median_filter def remove_baseline(ecg, fs360.0, window_ms200): 用中值濾波估計(jì)基線并扣除 window_ms 控制窗口大小200ms 對 QRS 太窄不會(huì)把波形吃掉 win int(fs * window_ms / 1000) // 2 * 2 1 # 強(qiáng)制奇數(shù)窗 baseline median_filter(ecg, sizewin, modenearest) return ecg - baselinewindow_ms這個(gè)參數(shù)我一般取 150–250ms。窗口小于 QRS 寬度時(shí)中值濾波會(huì)把 QRS 本身也當(dāng)成基線的一部分濾完波形就平了窗口太大又跟不上快速漂移。modenearest是為了避免邊緣效應(yīng)采樣點(diǎn)前段和中段的濾波效果差異明顯這類細(xì)節(jié)就是調(diào)參中的“血淚經(jīng)驗(yàn)”。2.3 濾波結(jié)果怎么驗(yàn)證頻響曲線與波形對照濾波不是跑完就完事的一定要做兩步驗(yàn)證。第一步是畫頻響曲線確認(rèn)通帶確實(shí)平坦、阻帶有足夠衰減import matplotlib.pyplot as plt from scipy.signal import sosfreqz b, a butter(4, [0.5 / 180.0, 50 / 180.0], btypeband) w, h sosfreqz([b], [a], worN2048, fs360) plt.semilogx(w, 20 * np.log10(abs(h)))第二步是直接把濾波前后的信號疊在一起畫。重點(diǎn)看兩件事R 波峰值是否被削平、濾波后的信號在起始段有沒有明顯上沖或下沖。filtfilt的邊緣效應(yīng)雖然比lfilter小但當(dāng)濾波器階數(shù)超過 5 時(shí)信號的頭尾幾十個(gè)采樣點(diǎn)仍然可能出現(xiàn)瞬時(shí)波動(dòng)。我習(xí)慣先把信號前后各延拓 2 秒再濾波濾完切掉徹底規(guī)避邊緣問題。3. 抓住 R 波Pan-Tompkins 實(shí)現(xiàn)、閾值自適應(yīng)與心率計(jì)算3.1 Pan-Tompkins 的四個(gè)階段與參數(shù)拆解MIT-BIH 注釋文件里的 R 波位置是人工逐搏校正過的算法上最貼近它精度的經(jīng)典方法還是 Pan-Tompkins。它分四步帶通濾波5–15Hz、差分、平方、滑動(dòng)窗口積分。帶通聚焦 QRS 的主要能量頻帶讓 P 波和 T 波大幅衰減差分突出 R 波的斜率特征平方讓所有值轉(zhuǎn)正并放大高頻分量滑動(dòng)窗口積分把單個(gè)尖峰變成一個(gè)光滑的駝峰方便用閾值判斷。def pan_tompkins(ecg, fs360.0, refractory0.2, win_ms150, thresh_factor0.35): 完整 Pan-Tompkins R 波檢測 返回 R 峰位置采樣點(diǎn)下標(biāo)和對應(yīng)的濾波信號 b, a butter(4, [5 / (fs / 2), 15 / (fs / 2)], btypeband) filtered filtfilt(b, a, ecg) diff np.diff(filtered, prependfiltered[0]) squared diff ** 2 win int(fs * win_ms / 1000) window np.ones(win) / win integrated np.convolve(squared, window, modesame) # 主角自適應(yīng)閾值 peak_thresh 0.3 * np.max(integrated[:int(fs * 2)]) r_peaks [] ref_period int(refractory * fs) i 0 while i len(integrated): if integrated[i] peak_thresh: # 以積分信號局部極大值為準(zhǔn)回找原始濾波信號上的最大點(diǎn) j min(i ref_period, len(integrated) - 1) seg integrated[i:j] loc i np.argmax(seg) r_loc i np.argmax(filtered[i:j]) r_peaks.append(r_loc) # 用最新峰值動(dòng)態(tài)更新閾值自適應(yīng)核心 peak_thresh 0.5 * peak_thresh 0.5 * integrated[loc] i j else: i 1 return np.array(r_peaks), filteredrefractory不應(yīng)期取 0.2 秒——正常心率下兩次 R 波間隔不會(huì)小于 0.2 秒這段時(shí)間內(nèi)即使有高幅噪聲也不會(huì)被當(dāng)作新搏動(dòng)。win_ms取 150ms 是經(jīng)驗(yàn)值它是 QRS 典型寬度 80–120ms 的 1.2–1.5 倍能讓積分窗口剛好覆蓋一個(gè)完整 QRS 駝峰。thresh_factor最初用前 2 秒信號最大值乘 0.3 作為初始閾值這段信號一般包含 2–4 個(gè)搏動(dòng)能適應(yīng)前 2 秒的幅值水平。這段代碼我做了個(gè)簡化處理初始閾值用固定比例等檢測到 5 個(gè)搏動(dòng)后再進(jìn)入“0.5 舊閾值 0.5 新峰值”的更新模式。這個(gè)策略在實(shí)際跑 MIT-BIH 的200.atr時(shí)表現(xiàn)不錯(cuò)但跑203.atr這類有大量室早的記錄時(shí)單閾值跟不上形態(tài)突變第二年就有論文提出雙閾值方案。后面避坑章節(jié)會(huì)展開。3.2 初始閾值失敗怎么辦雙閾值與回掃機(jī)制Pan-Tompkins 最經(jīng)典的翻車場景是前 2 秒信號里有個(gè)巨大偽差峰值閾值被抬得太高導(dǎo)致后續(xù)真正的 R 波全部漏檢或者前 2 秒全是低幅信號閾值太低后面高頻噪聲全被當(dāng)成 R 波。處理辦法是雙閾值回掃高閾值檢測出保真的 R 波集合對高閾值漏檢的片段用低閾值通常取高閾值的 0.5 倍重新掃一遍看能否找到時(shí)間位置合理的疑似峰。def detect_with_dual_threshold(integrated, filtered, fs, thresh_high0.3, thresh_low0.15, refractory0.2, methodpeak): 高閾值初篩 低閾值回掃緩解 R 波幅值突變導(dǎo)致的漏檢 thresh_high/thresh_low 可以傳絕對值也可以傳比例因子 rr_min int(refractory * fs) # 第一階段高閾值 high_peaks find_peaks(integrated, heightthresh_high, distancerr_min) # 第二階段每個(gè)相鄰高閾值峰之間的間隙用低閾值再掃 final_peaks list(high_peaks[0]) for k in range(len(final_peaks) - 1): gap_start final_peaks[k] rr_min gap_end final_peaks[k 1] - rr_min if gap_end - gap_start 0: continue seg integrated[gap_start:gap_end] local_max np.max(seg) if local_max thresh_low: loc_rel np.argmax(seg) final_peaks.append(gap_start loc_rel) final_peaks.sort() return final_peaks回掃不是無腦加峰兩個(gè)硬約束必須同時(shí)滿足候選峰與前后已確認(rèn)峰的距離必須大于不應(yīng)期回掃峰對應(yīng)的原始濾波信號幅值至少要達(dá)到前后 R 波平均幅值的三分之一。203.atr里室早和正常搏動(dòng)的形態(tài)差異極大高閾值抓到的是正常 QRS室早的積分幅值可能只有正常的一半這時(shí)候低閾值回掃就是救場的那個(gè)角色。3.3 心率計(jì)算的正確姿勢RR 間期異常值過濾R 波位置拿到之后心率計(jì)算本身不復(fù)雜但很容易算錯(cuò)。直接對所有 RR 間期取平均是一種常見錯(cuò)誤——因?yàn)槁z和誤檢會(huì)制造異常 RR 間期比如一次漏檢會(huì)把兩個(gè)搏動(dòng)合并成一個(gè) 2 秒的間隔一次誤檢會(huì)制造一個(gè) 0.1 秒的尖峰。正確的流程是先算全部 RR 間期把小于 0.4 秒或大于 1.5 秒的間期過濾掉再用剩余間期計(jì)算平均心率。import pandas as pd def compute_hr(r_peaks, fs360.0): 基于 R 峰位置計(jì)算平均心率和 RR 間期序列 rr np.diff(r_peaks) / fs # 單位秒 rr rr[(rr 0.4) (rr 1.5)] # 濾掉異常間隔 hr 60.0 / np.mean(rr) rr_df pd.DataFrame({rr_time: np.cumsum(rr), rr_value: rr}) return hr, rr_df濾波范圍 0.4–1.5 秒對應(yīng)心率 40–150 次/分鐘。健康成年人安靜狀態(tài)下心率基本在這個(gè)范圍里如果實(shí)測心率超過上限先把 R 波檢測的漏檢率檢查一遍——多數(shù)“爆表”的假心率都是漏檢導(dǎo)致的。這里pandas的使用不只是錦上添花后續(xù)做心率變異性HRV分析時(shí)RR 間期序列的時(shí)間對齊、缺失值填充、滾動(dòng)統(tǒng)計(jì)都需要 DataFrame 的接口。4. 特征提取與心律失常分類從 QRS 形態(tài)到隨機(jī)森林4.1 提取哪些特征QRS 幅度、寬度、QT 間期與 RR 間期搏動(dòng)級分類和心拍級分類用的特征集不同。心拍級分類區(qū)分正常、室早、房早關(guān)心的是單個(gè) QRS 的形態(tài)特征心律失常類型判斷房顫 vs 竇性關(guān)心的是 RR 間期的動(dòng)態(tài)特征。這套資源的核心是前者但它的特征提取函數(shù)是通用的完全可以直接復(fù)用。常見做法是以每個(gè) R 峰為中心往前取 100ms 估 Q 波起點(diǎn)往后取 300ms 估 T 波終點(diǎn)然后計(jì)算下面這組特征特征名計(jì)算方式臨床意義PeakAmpR 峰幅值 - Q 波谷幅值QRS 整體振幅QRSDurationS 波終點(diǎn) - Q 波起點(diǎn)正常 80–120ms增寬提示束支阻滯QTIntervalT 波終點(diǎn) - Q 波起點(diǎn)QT 延長與惡性心律失常相關(guān)PRIntervalR 峰前 120ms 處到 Q 波起點(diǎn)房室傳導(dǎo)時(shí)間RRPrev前一個(gè) RR 間期長度識別代償間歇RRAfter后一個(gè) RR 間期長度識別早搏后停頓T_AmpT 波峰值T 波倒置/高尖提示缺血def extract_beat_features(ecg, r_peaks, fs360.0): 以 R 峰為錨點(diǎn)在固定窗內(nèi)提取 QRS 形態(tài)特征 features [] q_win int(0.08 * fs) # Q 波搜索窗R 前 80ms s_win int(0.12 * fs) # S 波搜索窗R 后 120ms for r in r_peaks: if r - q_win 0 or r s_win len(ecg): continue q_min np.min(ecg[r - q_win:r]) s_min np.min(ecg[r:r s_win]) r_val ecg[r] peak_amp r_val - q_min qrs_dur (np.argmin(ecg[r:r s_win]) np.argmin(ecg[r - q_win:r])) / fs rr_prev (r_peaks[np.where(r_peaks r)[0][-1]] if any(r_peaks r) else r) - r if any(r_peaks r) else 0 features.append([peak_amp, qrs_dur, rr_prev, r_val, q_min, s_min]) return np.array(features)這些特征直接喂給分類器不夠還要做標(biāo)準(zhǔn)化。QRS 幅值跨度可以從 0.5mV 到 3mV而 PR 間期以毫秒為單位數(shù)值量級差幾百倍。sklearn的StandardScaler會(huì)在內(nèi)部把各特征拉到同一量綱這一步千萬別省否則 SVM 的核函數(shù)計(jì)算會(huì)被大數(shù)值特征主導(dǎo)小特征等于白提。4.2 SVM 與隨機(jī)森林的完整訓(xùn)練流程搏動(dòng)分類最穩(wěn)的組合是上述 6–8 維特征 隨機(jī)森林。隨機(jī)森林對特征量綱不敏感、對缺失值有容忍度而且能輸出特征重要性排序方便你反查哪些特征對分類貢獻(xiàn)最大。SVM 在小樣本高維場景下表現(xiàn)出色但需要調(diào)核函數(shù)和 C 參數(shù)如果只是為了快速拿到一個(gè)可用的分類器隨機(jī)森林是第一選擇。from sklearn.ensemble import RandomForestClassifier from sklearn.model_selection import train_test_split from sklearn.preprocessing import StandardScaler from sklearn.metrics import classification_report # X: (n_samples, n_features), y: 搏動(dòng)標(biāo)簽, 0正常, 1室早, 2房早 X_train, X_test, y_train, y_test train_test_split( X, y, test_size0.3, stratifyy, random_state42 ) scaler StandardScaler() X_train_scaled scaler.fit_transform(X_train) X_test_scaled scaler.transform(X_test) clf RandomForestClassifier( n_estimators300, max_depth10, class_weightbalanced, # 處理類別不均衡 random_state42 ) clf.fit(X_train_scaled, y_train) y_pred clf.predict(X_test_scaled) print(classification_report(y_test, y_pred))class_weightbalanced是必選項(xiàng)。MIT-BIH 的118系列記錄里正常搏動(dòng)可能占 90% 以上如果不加權(quán)分類器只要把所有樣本都預(yù)測為正常類就能拿到 90% 準(zhǔn)確率但室早一個(gè)都發(fā)現(xiàn)不了。stratifyy保證訓(xùn)練集和測試集里各類別比例與原數(shù)據(jù)一致避免隨機(jī)劃分時(shí)測試集恰好沒有某類樣本。4.3 房顫與室顫識別從 RR 間期變異到特征統(tǒng)計(jì)房顫的核心特征是 RR 間期高度不規(guī)律——絕對不齊。這個(gè)特性決定了它不需要太復(fù)雜的模型用 RR 間期序列的統(tǒng)計(jì)量就能做到相當(dāng)靠譜的初篩。我習(xí)慣提取三個(gè)指標(biāo)RR 間期標(biāo)準(zhǔn)差SDNN、連續(xù) RR 差值的均方根RMSSD、以及 Poincaré 散點(diǎn)圖的寬度——正常竇性心律的 Poincaré 圖呈彗星狀房顫則呈寬散形。def af_detection_features(rr_array): 房顫的 RR 間期特征SDNN、RMSSD、pNN50 rr np.asarray(rr_array, dtypefloat) sdnn np.std(rr) diff_rr np.abs(np.diff(rr)) rmssd np.sqrt(np.mean(diff_rr ** 2)) pnn50 np.mean(diff_rr 0.05) * 100 # 相鄰 RR 差超過 50ms 的占比 return np.array([sdnn, rmssd, pnn50])這三個(gè)特征喂給邏輯回歸或隨機(jī)森林對 MIT-BIH 里的房顫記錄比如105.atr附近的多條數(shù)據(jù)通常能拿到 0.85 以上的 AUC。室顫則是另一個(gè)模式信號退化成完全不規(guī)則的細(xì)碎波形R 波檢測基本失效這時(shí)候看的不再是 RR 間期而是信號的幅度概率密度和頻率分布。真遇到這個(gè)狀態(tài)算法層面已經(jīng)沒必要繼續(xù)走搏動(dòng)分類流程當(dāng)務(wù)之急是觸發(fā)惡性心律失常報(bào)警——這是個(gè)完全不同的檢測分支資源包里把它單列出來是有道理的。5. 避坑錦囊六個(gè)會(huì)翻車的 ECG 細(xì)節(jié)與排查手段5.1 R 波檢測結(jié)果里出現(xiàn)大量誤檢現(xiàn)象檢測出的 R 峰數(shù)量明顯多于實(shí)際搏動(dòng)數(shù)量且位置集中在 T 波或噪聲段。原因T 波在某些導(dǎo)聯(lián)上幅值不低經(jīng)過 5–15Hz 帶通后仍有殘留更常見的是滑動(dòng)窗口積分閾值設(shè)置過低T 波被當(dāng)成候選峰。還有一種隱蔽情況是refractory寫成了秒數(shù)忘記換算成采樣點(diǎn)數(shù)。解決先手動(dòng)畫圖把 R 峰標(biāo)注疊加在濾波后的信號上看誤檢峰與正常 R 波之間的間隔。如果誤檢峰與前面真峰的間隔小于 0.25 秒說明refractory失效或沒有生效如果是規(guī)律性出現(xiàn)在每個(gè) T 波位置把帶通低端從 5Hz 提到 8Hz衰減低斜率和 T 波的低頻分量。做完這兩步誤檢通常能降一個(gè)數(shù)量級。5.2 濾波后 QRS 形態(tài)整體變形現(xiàn)象濾波前后的 QRS 波形對不上R 波寬度變大或出現(xiàn)雙峰。原因帶通濾波器的order設(shè)置過高。階數(shù)超過 6 時(shí)濾波器在截止頻率附近會(huì)有嚴(yán)重的相位畸變雖然filtfilt補(bǔ)償了相位延遲但幅頻響應(yīng)的過沖會(huì)導(dǎo)致 QRS 邊緣出現(xiàn)振鈴偽跡。解決把order降回 4優(yōu)先保證波形形態(tài)完整性。如果對高頻噪聲的抑制不夠用級聯(lián)的方式先 4 階低通 50Hz再 4 階高通 0.5Hz兩級串聯(lián)對信號的相位影響比單級 8 階小得多。5.3 心率計(jì)算結(jié)果忽高忽低平均心率卻正常現(xiàn)象瞬時(shí)心率曲線劇烈跳變但平均心率在合理范圍。原因漏檢和誤檢同時(shí)存在。漏檢把兩個(gè) RR 間期合并成一個(gè)長間隔拉低瞬時(shí)心率誤檢又在里面插了一個(gè)短間隔把瞬時(shí)心率抬高。平均后正負(fù)抵消掩蓋了問題。解決瞬時(shí)心率曲線必須經(jīng)過中值濾波再展示。用長度為 5 的中值窗口能干凈利落地去掉單點(diǎn)異常同時(shí)回到 R 波檢測環(huán)節(jié)把漏檢的片段找出來單獨(dú)調(diào)閾值。我在實(shí)際項(xiàng)目里發(fā)現(xiàn)很多所謂的“心率算法不穩(wěn)定”問題根源不在心率計(jì)算而在 R 波檢測。5.4 ATR 文件讀進(jìn)來后與自己的檢測結(jié)果對不上現(xiàn)象用 WFDB 工具讀200.atr得到的 R 波位置和自己檢測出的位置總是差 5–10 個(gè)采樣點(diǎn)。原因MIT-BIH 的注釋位置標(biāo)注的是 QRS 波群的某個(gè)參考點(diǎn)通常是最大斜率點(diǎn)或峰值附近不同版本的標(biāo)注工具參考點(diǎn)定義不完全一致另一個(gè)常見原因是自己的信號預(yù)處理過程中用了np.diff沒有注意到diff會(huì)使信號長度減 1后續(xù)所有索引整體偏移。解決先確認(rèn)采樣率是否一致——MIT-BIH 是 360Hz如果你用自己的采樣率需要resample_poly做重采樣。然后允許檢測結(jié)果與注釋之間有一個(gè) ±15ms約 5 個(gè)采樣點(diǎn)的誤差窗口在這個(gè)窗口內(nèi)都記為正確檢出。臨床上評價(jià) QRS 檢測器的標(biāo)準(zhǔn)是靈敏度和陽性預(yù)測值本身就容忍 10ms 級的偏差。5.5 特征提取時(shí)切片越界程序崩潰現(xiàn)象提取特征時(shí)IndexError集中在信號尾部。原因最后一個(gè) R 峰距離信號末尾不足一個(gè)特征窗口長度向后取 S 波或 T 波窗口時(shí)溢出信號開頭的 R 峰則可能因?yàn)橄蚯叭?Q 波窗口導(dǎo)致負(fù)索引。解決特征提取函數(shù)里必須加邊界保護(hù)最粗暴但有效的方式是直接丟棄前后不足一個(gè)窗口長度的 R 峰。欠采樣時(shí)丟失幾個(gè)搏動(dòng)完全不影響分類器性能但越界崩潰會(huì)直接中斷整個(gè)流程。類似q_min np.min(ecg[r - q_win:r])這行代碼看起來簡單不加if r - q_win 0就是定時(shí)炸彈。5.6 訓(xùn)練集分類準(zhǔn)確率 95%測試集只有 60%現(xiàn)象模型在訓(xùn)練集上表現(xiàn)優(yōu)異獨(dú)立測試集上性能明顯下降。原因特征提取時(shí)使用了整個(gè)數(shù)據(jù)集的信息。最常見的泄漏是特征標(biāo)準(zhǔn)化用了全數(shù)據(jù)的均值和方差或者打亂訓(xùn)練集和測試集時(shí)沒有按記錄分組——同一條記錄里相鄰搏動(dòng)的特征高度相關(guān)被分到訓(xùn)練集和測試集兩側(cè)時(shí)模型相當(dāng)于“見過”了同類樣本。解決按記錄分組劃分?jǐn)?shù)據(jù)同一條記錄的所有搏動(dòng)只能出現(xiàn)在訓(xùn)練集或測試集中不能兩邊都出現(xiàn)。標(biāo)注數(shù)據(jù)時(shí)也要注意是否存在時(shí)間上的重疊窗口。這條規(guī)則適用于心電、腦電、肌電等所有時(shí)間序列相關(guān)的分類任務(wù)。6. 用 ATR 文件做驗(yàn)證把注釋解析成標(biāo)簽并量化檢測精度這套資源里的.atr文件是 MIT-BIH 的心律注釋文件它記錄了每個(gè)搏動(dòng)的類型代碼N 表示正常、V 表示室早、A 表示房早等和精確時(shí)間位置。對我們的價(jià)值是可以用它作為金標(biāo)準(zhǔn)量化自己 R 波檢測器的靈敏度和陽性預(yù)測值。import wfdb def load_atr_labels(record_name): 讀取 MIT-BIH 注釋文件返回搏動(dòng)位置與類型標(biāo)簽 ann wfdb.rdann(record_name, atr) beat_idx [i for i, s in enumerate(ann.symbol) if s in NLVARJES] beat_pos np.array(ann.sample)[beat_idx] beat_type np.array(ann.symbol)[beat_idx] return beat_pos, beat_type拿到金標(biāo)準(zhǔn)后匹配邏輯是這樣的對自己的每個(gè)檢測點(diǎn)在 ±15ms 容差范圍內(nèi)尋找最近的金標(biāo)準(zhǔn) R 波位置。找到則記為真陽性TP找不到則記為假陽性FP反過來金標(biāo)準(zhǔn)位置附近沒有自己的檢測點(diǎn)記為假陰性FN。靈敏度 TP/(TPFN) 和陽性預(yù)測值 TP/(TPFP) 都超過 0.95這算法才算合格。我的最后一步永遠(yuǎn)是可視化驗(yàn)證把 R 波峰和注釋位置同時(shí)畫在信號上用不同顏色區(qū)分真陽性、假陽性、漏檢。一套完整的心電分析流程從濾波到房顫識別中間任何一環(huán)出問題都會(huì)在圖上暴露出來。從那以后我每次跑新數(shù)據(jù)集都會(huì)強(qiáng)制走一遍這個(gè)流程——先把濾波前后波形疊畫檢查再跑 R 波檢測最后對標(biāo)簽算指標(biāo)缺一步都不安心。希望幫到你。本文還有配套的精品資源點(diǎn)擊獲取