H2O简单深度/机器学习框架介绍(一)

2020-12-28 14:35:43 浏览数 (1)

今天看到cell report上面的一篇ML/DL的文章竟然是用的H2O的框架,没有用更常见的tensorflow/pytorch, 去查了一下,发现这也是个新框架(2014年),而且用的人也不少,而且最重要的是,比较简单,不过感觉缺乏一定的灵活性。

H2O是一个基于java的机器学习/深度学习平台,它支持大量无监督和有监督的模型,也支持深度学习算法;可以作为R或Python包导入,也给用户提供UI似的界面。

在R中可直接下载安装:

代码语言:javascript复制
install.packages('h2o')
library(h2o)

在Python中也可以直接导入:

代码语言:javascript复制
import h2o

导入之后第一步需要初始化:

代码语言:javascript复制
h2o.init(nthreads = 2) 

会返回cluster的信息:

训练和validate的步骤也都是一步到位,之后test一下再接着写写。有兴趣做DL方面,但是又不想去学pytorch这种框架的可以考虑去用一用。

0 人点赞