neurojs
neurojs 是一个用于在浏览器中进行深度学习的 JavaScript 框架。它主要关注强化学习,但也可以用于任何基于神经网络的任务。它包含了一些精美的演示,可以可视化这些功能,例如一个 2D 自驾车。
欢迎贡献。在社区中开发这样的项目更加令人兴奋!
注意:自从更通用的框架如 TensorFlow-JS 出现后,这个仓库已经不再维护,我建议你使用这些框架。
特性
- 实现了一个全栈的基于神经网络的机器学习框架
- 扩展的强化学习支持
- 统一和优先重放缓冲区
- 增强学习(增加行动间隙)https://arxiv.org/pdf/1512.04860v1.pdf
- 支持 deep-q-networks 和 actor-critic 模型(通过深度确定性策略梯度)
- 网络配置的二进制导入和导出(权重等)
- 高性能
示例
- 自驾车
- 高级 XOR
- Andrej Karpathy 的 Waterworld (ConvNetJS 被 NeuroJS 替代)
运行示例
npm install
npm start
在浏览器中打开 http://localhost:8080/examples/
,并选择你想运行的演示。
接下来是什么?
- 更多示例(乒乓、摆锤、贪吃蛇、改进版自驾车等)
- 支持 web workers
- LSTM 和通过时间的反向传播