< 返回版块

Lyn-liyuan 发表于 2019-08-24 00:18

Tags:ANN,深度学习,人工智能

学了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

评论区

写评论
作者 Lyn-liyuan 2019-08-24 17:01

手快多打了一个i,谢谢指正 对以下内容的回复:

shaitao 2019-08-24 14:55

Softimax/ 捂脸

Mike Tang 2019-08-24 13:16

厉害的。

damody 2019-08-24 08:47

還蠻厲害的

1 共 4 条评论, 1 页