首页    期刊浏览 2024年09月20日 星期五
登录注册

文章基本信息

  • 标题:OmniGA: Optimized Omnivariate Decision Trees for Generalizable Classification Models
  • 本地全文:下载
  • 作者:Arturo Magana-Mora ; Vladimir B. Bajic
  • 期刊名称:Scientific Reports
  • 电子版ISSN:2045-2322
  • 出版年度:2017
  • 卷号:7
  • 期号:1
  • DOI:10.1038/s41598-017-04281-9
  • 语种:English
  • 出版社:Springer Nature
  • 摘要:Classification problems from different domains vary in complexity, size, and imbalance of the number of samples from different classes. Although several classification models have been proposed, selecting the right model and parameters for a given classification task to achieve good performance is not trivial. Therefore, there is a constant interest in developing novel robust and efficient models suitable for a great variety of data. Here, we propose OmniGA, a framework for the optimization of omnivariate decision trees based on a parallel genetic algorithm, coupled with deep learning structure and ensemble learning methods. The performance of the OmniGA framework is evaluated on 12 different datasets taken mainly from biomedical problems and compared with the results obtained by several robust and commonly used machine-learning models with optimized parameters. The results show that OmniGA systematically outperformed these models for all the considered datasets, reducing the F1 score error in the range from 100% to 2.25%, compared to the best performing model. This demonstrates that OmniGA produces robust models with improved performance. OmniGA code and datasets are available at www.cbrc.kaust.edu.sa/omniga/.
国家哲学社会科学文献中心版权所有