This repository is used for experiments considering the Regression Neural Gas and Regression-Sensitive Neural Gas [source].
There are also other models implemented: Radial-Basis-Function Network ([RBFN]), Regression Learning Vector Quantization ([RLVQ]), Neural Gas for time-series prediction ([NGTSP]) and a variant of it, in which the predictor is altered to be dependent on the data samples rather than on the distance vector (xNGTSP / RNGTSP)
In the file experiments.py the code can be found used to produce the results of a comparison of these models. This section deals with the setting. A comprehensive overview of the results is available in the folder results including a summary.
The datasets included are WineQuality-red, California Housing, Breastcancer Prognostics and Diabetes. All Datasets are normalized in range [0, 1]. Furthermore, different targets can be chosen for Winequality and Breastcancer. For Winequality we chose for experiments.py the target alcohol. For Breastcancer we went with the mean perimeter as the target and removed the columns ID and Outcome (for the sake of normalization) and the column Lymph Node Status, due to missing values.
All models were initialized via
for the RBFN with prototype/center
for RegNG and RegSeNG.
Further for the parameter
Furthermore, a batch-normlization layer was applied to accelerate training and enhance reproducability.
We used a 5-fold Cross-Validation for each 5, 10 and 15 prototypes. For validation measures we used the coefficient of determination
Note that there is an additional measure
In the summary.csv also the maximum achieved values for each measure are recorded.