今天看到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这种框架的可以考虑去用一用。