10 个版本

0.3.2	2024 年 1 月 13 日
0.3.1	2024 年 1 月 6 日
0.2.4	2023 年 12 月 31 日
0.1.2	2023 年 12 月 10 日

#342 in 机器学习

每月 116 次下载

MIT 许可证

28KB
373 行

极小学习机（ELM）

极小学习机（ELM）crate。一个最小化且灵活的crate，可用于训练ELM，一种神经网络。目前支持单隐藏层和回归任务。

参考文献

原始论文：https://ieeexplore.ieee.org/document/1380068
维基百科：https://en.wikipedia.org/wiki/Extreme_learning_machine

基本用法

use elm::{ELM, Epsilon};
use elm::activation_functions::ActivationFunction;

let mut elm = ELM::new(2, 4, 2, ActivationFunction::LeakyReLU, Epsilon::Default);
let inputs: Vec<Vec<f64>> = vec![vec![1.0, 0.0], vec![1.0, 0.0]];
let targets: Vec<Vec<f64>> = vec![vec![1.0, 1.0], vec![1.0, 1.5]];
elm.train(&inputs, &targets);

let new_inputs: Vec<Vec<f64>> = vec![vec![1.0, 4.0], vec![1.3, 0.6]];
let prediction = elm.predict(&new_inputs);