This paper presents a model to predict the risk of depression based on electrocardiogram (ECG). This proposed model uses a Recurrent Neural Network (RNN) and Long Short-Term Memory (LSTM) autoencoder to predict normal, abnormal, and PVC heartbeats. The RNN model is a deep learning-based model to classify normal, abnormal, and PVC heartbeats. We used the model as a classifier. The model uses a heart rates dataset to predict abnormal and PVC heartbeats. As for the dataset, we have used 5000 ECG samples. The model was trained on a training dataset and validation dataset. After that, it was tested on a test dataset. The model is trained on normal heartbeat rates, so the model can predict any heartbeat rates other than normal. Our contribution here is to build a model that can differentiate between “normal,” “abnormal,” and “risky” heartbeats. Our model predicts “normal” heartbeats with 97.24% accuracy and can predict “PVC” heartbeats with 100% accuracy. Other than the accuracy, we evaluated our model on the training loss graphs. These two types of training loss graphs were evaluated as “normal” versus “risky” and “abnormal” versus “risky.” We have seen great results there as well. The best losses for “normal,” “abnormal,” and “risky” are 5.71, 33.36, and 34.78. However, these results may improve if a larger dataset is used. In studies, it was found that patients suffering from depression may have a different kind of heartbeat than the normal ones. In most cases, it is PVC (Premature Ventricular Contraction) heartbeats. Therefore, the target is to predict abnormal heartbeats and PVC heartbeats.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.
customersupport@researchsolutions.com
10624 S. Eastern Ave., Ste. A-614
Henderson, NV 89052, USA
This site is protected by reCAPTCHA and the Google Privacy Policy and Terms of Service apply.
Copyright © 2024 scite LLC. All rights reserved.
Made with 💙 for researchers
Part of the Research Solutions Family.