# jstarcraft-ai **Repository Path**: TM16/jstarcraft-ai ## Basic Information - **Project Name**: jstarcraft-ai - **Description**: 目标是提供一个完整的Java机器学习(Machine Learning/ML)框架,作为人工智能在学术界与工业界的桥梁. 让相关领域的研发人员能够在各种软硬件环境/数据结构/算法/模型之间无缝切换. 涵盖了从数据处理到模型的训练与评估各个环节,支持硬件加速和并行计算,是最快最全的Java机器学习库. - **Primary Language**: Unknown - **License**: Apache-2.0 - **Default Branch**: master - **Homepage**: None - **GVP Project**: No ## Statistics - **Stars**: 0 - **Forks**: 4 - **Created**: 2022-09-07 - **Last Updated**: 2022-09-07 ## Categories & Tags **Categories**: Uncategorized **Tags**: None ## README # JStarCraft AI **** [![License](https://img.shields.io/badge/license-Apache%202-4EB1BA.svg)](https://www.apache.org/licenses/LICENSE-2.0.html) [![Total lines](https://tokei.rs/b1/github/HongZhaoHua/jstarcraft-ai?category=lines)](https://tokei.rs/b1/github/HongZhaoHua/jstarcraft-ai?category=lines) [![Codacy Badge](https://api.codacy.com/project/badge/Grade/8e39a24e1be740c58b83fb81763ba317)](https://www.codacy.com/project/HongZhaoHua/jstarcraft-ai/dashboard?utm_source=github.com&utm_medium=referral&utm_content=HongZhaoHua/jstarcraft-ai&utm_campaign=Badge_Grade_Dashboard) 希望路过的同学,顺手给JStarCraft框架点个Star,算是对作者的一种鼓励吧! **** **JStarCraft AI是一个机器学习的轻量级框架.遵循Apache 2.0协议.** 在学术界,绝大多数研究人员使用的编程语言是Python. 在工业界,绝大多数开发人员使用的编程语言是Java. JStarCraft AI是一个基于Java语言的机器学习工具包,由一系列的数据结构,算法和模型组成. 目标是作为在学术界与工业界从事机器学习研发的相关人员之间的桥梁.普及机器学习在Java领域的应用. |作者|洪钊桦| |---|--- |E-mail|110399057@qq.com, jstarcraft@gmail.com **** ## JStarCraft AI架构 JStarCraft AI框架各个模块之间的关系: ![ai](https://github.com/HongZhaoHua/jstarcraft-tutorial/blob/master/ai/JStarCraft%E4%BA%BA%E5%B7%A5%E6%99%BA%E8%83%BD%E6%A1%86%E6%9E%B6%E7%BB%84%E4%BB%B6%E5%9B%BE.png "JStarCraft AI架构") **** ## JStarCraft AI特性 * [1.数据(data)](https://github.com/HongZhaoHua/jstarcraft-ai/wiki/%E6%95%B0%E6%8D%AE) * 属性与特征 * 连续 * 离散 * 模块与实例 * 选择,排序与切割 * 2.环境(environment) * 串行计算 * 并行计算 * CPU计算 * GPU计算 * 3.数学(math) * 算法(algorithm) * 微积分(calculus) * 相关性(correlation) * 距离(distance) * 相似度(similarity) * 分解(decomposition) * 核技巧(kernel) * 概率(probability) * [数据结构(structure)](https://github.com/HongZhaoHua/jstarcraft-ai/wiki/%E6%95%B0%E6%8D%AE%E7%BB%93%E6%9E%84) * 标量 * 向量 * 矩阵 * 张量 * 单元 * 表单 * 4.调制解调(modem) * [5.模型(model)](https://github.com/HongZhaoHua/jstarcraft-ai/wiki/%E6%A8%A1%E5%9E%8B) * 线性模型(linear) * 近邻模型(nearest neighbor) * 矩阵分解模型(matrix factorization) * 神经网络模型(neutral network) * 计算图 * 节点 * 层 * 正向传播与反向传播 * 激活函数 * 梯度更新 * 概率图模型(probabilistic graphical) * 规则模型(rule) * 支持向量机模型(support vector machine) * 树模型(tree) * 6.优化(optimization) * 梯度下降法(gradient descent) * 批量梯度下降(batch gradient descent) * 随机梯度下降(stochastic gradient descent) * 牛顿法和拟牛顿法(newton method/quasi newton method) * 共轭梯度法(conjugate gradient) * [试探法(heuristic)](https://github.com/HongZhaoHua/jstarcraft-ai/wiki/%E8%AF%95%E6%8E%A2%E6%B3%95) * 模拟退火算法(simulate anneal) * 遗传算法(genetic) * 蚁群算法(ant colony) * 粒子群算法(particle swarm) * [7.有监督学习(supervised)](https://github.com/HongZhaoHua/jstarcraft-ai/wiki/%E6%9C%89%E7%9B%91%E7%9D%A3%E5%AD%A6%E4%B9%A0) * 分类 * 回归 * [8.无监督学习(unsupervised)](https://github.com/HongZhaoHua/jstarcraft-ai/wiki/%E6%97%A0%E7%9B%91%E7%9D%A3%E5%AD%A6%E4%B9%A0) * 聚类 * 关联 * [9.丰富的评估指标](#评估指标) * [排序指标](#排序指标) * [评分指标](#评分指标) **** ## JStarCraft AI教程 * 1.设置依赖 * [Maven依赖](#Maven依赖) * [Gradle依赖](#Gradle依赖) * 2.配置环境 * [设置CPU环境](#设置CPU环境) * [设置GPU环境](#设置GPU环境) * [使用环境上下文](#使用环境上下文) * 3.使用数据 * [数据表示](#数据表示) * [数据转换](#数据转换) * ARFF * CSV * JSON * Parquet * HQL * SQL * [数据处理](#数据处理) * 选择 * 排序 * 切割 #### Maven依赖 ```maven com.jstarcraft ai 1.0 ``` #### Gradle依赖 ```gradle compile group: 'com.jstarcraft', name: 'ai', version: '1.0' ``` #### 设置CPU环境 ```maven org.nd4j nd4j-native-platform 1.0.0-beta3 ``` #### 设置GPU环境 * CUDA 9.0 ```maven org.nd4j nd4j-cuda-9.0-platform 1.0.0-beta3 ``` * CUDA 9.1 ```maven org.nd4j nd4j-cuda-9.1-platform 1.0.0-beta3 ``` * CUDA 9.2 ```maven org.nd4j nd4j-cuda-9.2-platform 1.0.0-beta3 ``` * CUDA 10.0 ```maven org.nd4j nd4j-cuda-10.0-platform 1.0.0-beta3 ``` * CUDA 10.1 ```maven org.nd4j nd4j-cuda-10.1-platform 1.0.0-beta3 ``` #### 使用环境上下文 ```java // 获取默认环境上下文 EnvironmentContext context = EnvironmentContext.getContext(); // 在环境上下文中执行任务 Future task = context.doTask(() - > { int dimension = 10; MathMatrix leftMatrix = getRandomMatrix(dimension); MathMatrix rightMatrix = getRandomMatrix(dimension); MathMatrix dataMatrix = getZeroMatrix(dimension); dataMatrix.dotProduct(leftMatrix, false, rightMatrix, true, MathCalculator.PARALLEL); }); ``` #### 数据表示 * 未处理的形式(转换前) | 用户(User) | 旧手机类型(Item) | 新手机类型(Item) | 评分(Score) | | :----: | :----: | :----: | :----: | | Google Fan | Android | Android | 3 | | Google Fan | Android | IOS | 1 | | Google Fan | IOS | Android | 5 | | Apple Fan | IOS | IOS | 3 | | Apple Fan | Android | IOS | 5 | | Apple Fan | IOS | Android | 1 | * 已处理的形式(转换后) | 定性(User) | 定性(Item) | 定性(Item) | 定量(Score) | | :----: | :----: | :----: | :----: | | 0 | 0 | 0 | 3 | | 0 | 0 | 1 | 1 | | 0 | 1 | 0 | 5 | | 1 | 1 | 1 | 3 | | 1 | 0 | 1 | 5 | | 1 | 1 | 0 | 1 | #### 数据转换 **数据转换器**(DataConverter)负责各种各样的格式转换为JStarCraft AI框架能够处理的**数据模块**(DataModule). JStarCraft AI框架各个转换器与其它系统之间的关系: ![converter](https://github.com/HongZhaoHua/jstarcraft-tutorial/blob/master/ai/%E8%BD%AC%E6%8D%A2%E5%99%A8%E7%B1%BB%E5%9B%BE.png "转换器") * 定义数据属性 ```java // 定性属性 Map> qualityDifinitions = new HashMap<>(); qualityDifinitions.put("user", String.class); qualityDifinitions.put("item", String.class); // 定量属性 Map> quantityDifinitions = new HashMap<>(); quantityDifinitions.put("score", float.class); DataSpace space = new DataSpace(qualityDifinitions, quantityDifinitions); ``` * 定义数据模块 ```java TreeMap configuration = new TreeMap<>(); configuration.put(1, "user"); configuration.put(3, "item"); configuration.put(4, "score"); DataModule module = space.makeDenseModule("module", configuration, 1000); ``` **JStarCraft AI框架兼容的格式** * ARFF ```java // ARFF转换器 ArffConverter converter = new ArffConverter(space.getQualityAttributes(), space.getQuantityAttributes()); // 获取流 File file = new File(this.getClass().getResource("module.arff").toURI()); InputStream stream = new FileInputStream(file); // 转换数据 int count = converter.convert(module, stream, null, null, null); ``` * CSV ```java // CSV转换器 CsvConverter converter = new CsvConverter(',', space.getQualityAttributes(), space.getQuantityAttributes()); // 获取流 File file = new File(this.getClass().getResource("module.csv").toURI()); InputStream stream = new FileInputStream(file); // 转换数据 int count = converter.convert(module, stream, null, null, null); ``` * JSON ```java // JSON转换器 JsonConverter converter = new JsonConverter(space.getQualityAttributes(), space.getQuantityAttributes()); // 获取流 File file = new File(this.getClass().getResource("module.json").toURI()); InputStream stream = new FileInputStream(file); // 转换数据 int count = converter.convert(module, stream, null, null, null); ``` * HQL ```java // HQL转换器 QueryConverter converter = new QueryConverter(space.getQualityAttributes(), space.getQuantityAttributes()); // 获取游标 String selectDataHql = "select data.user, data.leftItem, data.rightItem, data.score from MockData data"; Session session = sessionFactory.openSession(); Query query = session.createQuery(selectDataHql); ScrollableResults iterator = query.scroll(); // 转换数据 int count = converter.convert(module, iterator, null, null, null); session.close(); ``` * SQL ```java // SQL转换器 QueryConverter converter = new QueryConverter(space.getQualityAttributes(), space.getQuantityAttributes()); // 获取游标 String selectDataSql = "select user, leftItem, rightItem, score from MockData"; Session session = sessionFactory.openSession(); Query query = session.createQuery(selectDataSql); ScrollableResults iterator = query.scroll(); // 转换数据 int count = converter.convert(module, iterator, null, null, null); session.close(); ``` #### 数据处理 * 选择 ```java ``` * 排序 ```java ``` * 切割 ```java ``` **** ## 评估指标 #### 排序指标 - AUC - Diversity - MAP - MRR - NDCG - Novelty - Precision - Recall #### 评分指标 - MAE - MPE - MSE/RMSE