University of Wollongong
Browse

File(s) not publicly available

PSPSO: A package for parameters selection using particle swarm optimization

journal contribution
posted on 2024-11-17, 16:54 authored by Ali Haidar, Matthew Field, Jonathan Sykes, Martin Carolan, Lois Holloway
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.

Funding

Western Sydney Local Health District (NSW 2170)

History

Journal title

SoftwareX

Volume

15

Language

English

Usage metrics

    Categories

    No categories selected

    Exports

    RefWorks
    BibTeX
    Ref. manager
    Endnote
    DataCite
    NLM
    DC