In this work, a novel method for selecting the optimal set of input features for classifying the presence of congestive heart failure (CHF) using a supervised machine learning approach is presented. A random forest classifier (RFC) was utilized to carry out the binary classification task and two different models were explored. We employed the embedded RFC feature importance attribute for the first model, and a multi-classifier technique which integrates the feature weight importance correlation (F-WIC) method was adopted for the second model. Our results show that the second model using the F-WIC method offers superior performance (100% accuracy) and provides a generalized approach to feature engineering for machine learning models irrespective of the algorithm used. This work offers a novel method for selecting the optimal set of input features for classifying the presence of congestive heart failure (CHF) using a machine learning approach.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Optimizing Classification of Congestive Heart Failure Using Feature Weight Importance Correlation

  • Iyabosola B. Oronti,
  • Muhammad S. Haleem,
  • Leandro Pecchia,
  • Thomas Popham

摘要

In this work, a novel method for selecting the optimal set of input features for classifying the presence of congestive heart failure (CHF) using a supervised machine learning approach is presented. A random forest classifier (RFC) was utilized to carry out the binary classification task and two different models were explored. We employed the embedded RFC feature importance attribute for the first model, and a multi-classifier technique which integrates the feature weight importance correlation (F-WIC) method was adopted for the second model. Our results show that the second model using the F-WIC method offers superior performance (100% accuracy) and provides a generalized approach to feature engineering for machine learning models irrespective of the algorithm used. This work offers a novel method for selecting the optimal set of input features for classifying the presence of congestive heart failure (CHF) using a machine learning approach.