学了1周左右RUST,在github创建了一个项目不依赖任何第三方库,自己手工编写人工神经网络网络库。现在已经可以跑用全连接层训练iris数据集。支持全连接层,卷积层,ReLu,Softmax,Sigoimd.
使用起来是这样的
let mut model = Model![
Dense!(8,4),
ReLu!(),
Dense!(3,8),
Softmax!()
];
let (data, labels) = load_data(args.path.as_path());
for p in 0..6000 {
println!("epoch {} loss = {}", p, 1f32 - model.fit(&data, &labels, 0.01f32));
}
https://github.com/Lyn-liyuan/hetu
1
共 4 条评论, 1 页
评论区
写评论手快多打了一个i,谢谢指正 对以下内容的回复:
Softimax/ 捂脸
厉害的。
還蠻厲害的