PSPSO: A package for parameters selection using particle swarm optimization
This paper reports a high-level python package for selecting machine learning algorithms and ensembles of machine learning algorithms parameters by using the particle swarm optimization (PSO) algorithm named PSPSO. The first version of PSPSO supports four algorithms: Support Vector Machine (SVM), Mu...
Gespeichert in:
Veröffentlicht in: | SoftwareX 2021-07, Vol.15, p.100706, Article 100706 |
---|---|
Hauptverfasser: | , , , , |
Format: | Artikel |
Sprache: | eng |
Schlagworte: | |
Online-Zugang: | Volltext |
Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
Zusammenfassung: | This paper reports a high-level python package for selecting machine learning algorithms and ensembles of machine learning algorithms parameters by using the particle swarm optimization (PSO) algorithm named PSPSO. The first version of PSPSO supports four algorithms: Support Vector Machine (SVM), Multi-Layer Perceptron (MLP), Extreme Gradient Boosting (XGBoost) and Gradient Boosting Decision Trees (GBDT). PSPSO provides an easy framework for building machine learning algorithms using PSO and a new platform for researchers to investigate their selection methods. In addition, it provides a basis for establishing new selection ideas and can be easily extended to support other algorithms. |
---|---|
ISSN: | 2352-7110 2352-7110 |
DOI: | 10.1016/j.softx.2021.100706 |