4个版本
0.2.0 | 2023年1月23日 |
---|---|
0.1.2 | 2022年4月1日 |
0.1.1 | 2022年3月19日 |
0.1.0 | 2022年3月15日 |
#834 in 机器学习
215KB
1.5K SLoC
Brains
注意:此项目非常早期,起源于我自己的架构研究。欢迎贡献、意见和问题!1.X.X版本将视为可用于公共使用的版本。
基于TensorFlow-rs绑定构建的人工神经网络框架,可以创建类似keras的架构,同时也可以在低级Tensorflow中直接集成自定义层。包括本地检查点、推理、批量训练和迭代训练。请参阅单元测试以了解事物如何调用,直到创建文档。另外,如果您想卸载计算,请确保启用tensorflow的GPU标志。
目前所有输入和输出都表示为扁平化的1D rust Vecs。
一旦以下TODO完成,此项目将在crates上更新,并可以被视为一个初级框架(v1.X.X)
TODO
重构所有内容以使用特性和构建器模式,以获得更类似于Keras的顺序API用户界面,以及可以更轻松地构建和在其他框架(如libtorch)上实现的东西。
创建一个dev分支并将其拉入Master作为发布分支
文档。文档字符串和项目概述页面,讨论Mod。
请告诉我您的看法: [email protected]
依赖项
~25MB
~546K SLoC