转载

码云推荐 | 基于 Java 的深度学习工具箱 ml4ai

ml4ai

Machine Learning for Artificial Intelligence

项目介绍

机器学习,人工智能,张量计算库。致力于开发一个基于java语言的深度学习工具箱,项目是基于Nd4j做运算,可以使用GPU加速大矩阵。支持自动微分、动态计算图等。 开发这个的目的在于结合java与pytorch的优势,目前来说还没有一个很易于编程的基于java的自动微分张量计算图工具,本工具就是为了此编写。

软件架构

软件架构说明

依赖张量运算库:Nd4j 可选的库:dl4j

安装教程

  1. 目前以提供源码的形式提供库,注意项目依赖nd4j,在pom中请引入相关库。

使用说明

1、提供了实例,以实例来做为教程,请原谅没有时间做视频或者专业资源,提供了实例来说明。

动量实例:com.ml4ai.demo.MomentTest

自动微分:com.ml4ai.demo.AutoGrad

MLP分类:com.ml4ai.demo.MLPClassification

拟合kx plus b:com.ml4ai.demo.Demo

参与

请联系: sleechengn@163.com ,加入力量,这是一个持续开发的项目,希望大家为java深度学习做贡献。

网站 :www.ml4ai.com

原文  https://gitee.com/sleechengn/ml4ai
正文到此结束
Loading...