elm_train.m

最新推荐文章于 2024-01-31 10:33:17 发布

CV小公举

最新推荐文章于 2024-01-31 10:33:17 发布

阅读量992

点赞数

分类专栏： matlab 文章标签： elm

本文链接：https://blog.csdn.net/qq_34685728/article/details/79117846

版权

elm_train.m 是一个用于训练极端学习机（ELM）的函数，支持回归和分类任务。用户可以指定数据文件、ELM类型、隐藏神经元数量和激活函数。函数返回训练时间和训练精度。支持Sigmoid、Sine和Hardlim三种激活函数，并自动设置输出神经元数量以适应多类别分类。函数还保存训练模型以便后续使用。

摘要由CSDN通过智能技术生成

function [TrainingTime,TrainingAccuracy] = elm_train(TrainingData_File, Elm_Type, NumberofHiddenNeurons, ActivationFunction)

% Usage: elm_train(TrainingData_File, Elm_Type, NumberofHiddenNeurons, ActivationFunction)
% OR:    [TrainingTime, TrainingAccuracy] = elm_train(TrainingData_File, Elm_Type, NumberofHiddenNeurons, ActivationFunction)
%
% Input:
% TrainingData_File     - Filename of training data set
% Elm_Type              - 0 for regression; 1 for (both binary and multi-classes) classification
% NumberofHiddenNeurons - Number of hidden neurons assigned to the ELM
% ActivationFunction    - Type of activation function:
%                           'sig' for Sigmoidal function
%                           'sin' for Sine function
%                           'hardlim' for Hardlim function
%
% Output:
% TrainingTime          - Time (seconds) spent on training ELM
% TrainingAccuracy      - Training accuracy:
%                           RMSE for regression or correct classification rate for classification
%
% MULTI-CLASSE CLASSIFICATION: NUMBER OF OUTPUT NEURONS WILL BE AUTOMATICALLY SET EQUAL TO NUMBER OF CLASSES
% FOR EXAMPLE, if there are 7 classes in all, there will have 7 output
% neurons; neuron