6 个版本

0.2.3 2022 年 9 月 29 日
0.2.2 2022 年 9 月 29 日
0.1.1 2022 年 9 月 23 日

662机器学习

MIT 许可证

11KB
277

该库可以创建非常基础的卷积神经网络。学习功能尚未实现。库仍需大量工作。

在 brainy 中,一个 layer 结构代表神经网络的一层(不包括输出层)以及指向下一层的映射。目前激活函数被视为与矩阵乘法等操作分开的映射。这样,可以“混合和匹配”不同的映射。

有关底层数学的描述和代码的更深入解释,请见以下链接

https://thefrogblog.xyz/neural-network-1/

https://thefrogblog.xyz/neural-network-2/

https://thefrogblog.xyz/neural-network-3/

一些示例代码

extern crate brainy;
use brainy::matrix::Matrix;
use brainy::layer::Map;
use brainy::network::Network;

fn main()
{
let mut network = Network::new();
network.append_layer(Map::MatrixMultiply(4,3));
network.append_layer(Map::Sigmoid(3));
network.append_layer(Map::MatrixMultiply(3,2));

let input = vec![0.0, 1.0, 1.0, 0.5];
let x = Matrix { rows: input.len(),  cols: 1, elements: input };

let y = network.feedforward(x);
y.print();
}

依赖项

~310KB