ITPub博客

首页 > 应用开发 > > 「java版本」机器学习工具包

「java版本」机器学习工具包

作者:coyan 时间:2019-09-27 10:16:07 0 删除 编辑

IT技术百货 



今天跟大家分享一个java版本的机器学习算法及工具包,内容非常的丰富;

java版本的优势

目前机器学习算法主要以python语言实现为主,但实际生产环境的服务往往以java为主。所以需要通过RPC调用、中间转换层(将模型转为java版,比如pb文件)或者是实现java版的前向计算等方式,供java服务使用;模型的改动、部署会比较耗时、费力;

基于Java的机器学习算法模型可以直接由java服务调用,不需要中间层转换等逻辑,部署和使用非常方便

目前大数据生态组件对java系语言的支持是最好的,比如spark、hadoop等;面对大数据处理时,python的优势不及java;

工具包介绍

项目位于github:https://github.com/luckyPT/jvm-ml,

所有的算法和工具都是java版本,主要提供了以下算法以及工具:

机器学习算法:

逻辑回归、决策树(CART树)、随机森林、GBDT、XGBOOST(包括单机版本以及spark版本)、线性支撑向量机(SVM)、fastText训练词向量以及文本分类、贝叶斯、经典的全连接神经网络、卷积神经网络、循环神经网络、基于深度学习的词向量训练;

DEMO目录

数据处理工具:

spark生产tf-record格式文件、Ansj分词、JieBa分词、数据的标准化与归一化处理、TF-IDF计算、连续特征离散化、编辑距离计算、PCA降维、TNSE降维、LDA聚类;

效果评估及可视化:

二分类的ROC曲线、PR曲线、阈值与PR曲线、阈值F1曲线绘制;

多分类评估指标计算,精确度、加权准确率召回率、F1值、各类别的准确率-召回率-F1值;

roc曲线

阈值 - PR曲线

图像绘制工具包:

支持绘制散点图、曲线、柱状图;

柱状图

项目地址:https://github.com/luckyPT/jvm-ml

来自 “ ITPUB博客 ” ,链接:https://blog.itpub.net/2236/viewspace-3133757/ ,如需转载,请注明出处,否则将追究法律责任。

请登录后发表评论 登录
全部评论

注册时间:2019-07-11

  • 博文量
    271
  • 访问量
    16167