niiknow/niilearn

镜像概述

niilearn是一个基于Node.js开发的机器学习REST API镜像，实现朴素贝叶斯算法，提供数据分类、模型训练等功能，可作为独立微服务部署，支持数据持久化。

核心功能

• 朴素贝叶斯分类：自动识别文本等数据的类别（如商品分类、语言、性别等）
• 持续训练：通过JSON或CSV格式补充训练数据，提升模型输出质量
• 模型重置：清空指定模型数据，重新开始训练
• 独立微服务：Docker化部署，可作为SaaS风格的机器学习API使用
• 数据持久化：挂载/app/data卷保存模型数据，避免容器重启丢失

使用场景

• 电商平台商品自动分类
• 文本内容的语言识别
• 用户数据的性别/兴趣标签自动生成
• 其他需要批量分类的业务场景

配置说明

• 数据持久化：将本地目录挂载到容器/app/data路径，例如-v ./local-niilearn-data:/app/data

API接口

POST /api/v1/bayes/train/:modelName

提交JSON格式的数据集和标签训练模型：

json
{
    "dataset": ["string1", "string2", ..., "stringn"],
    "labels": ["cat1", "cat2", ..., "catn"]
}

POST /api/v1/bayes/train-csv/:modelName?label=fieldName&text=fieldName

提交CSV格式数据训练（需含表头，可通过query参数指定文本和标签字段）

POST /api/v1/bayes/reset/:modelName

清空指定模型的训练数据

POST /api/v1/bayes/classify/:modelName

批量或单条分类数据：

json
{
    "dataset": ["sometext1", "sometext2"]
}

返回结果：

json
{
    "result": ["cat1", "cat2"]
}

部署示例

Docker Run命令

bash
docker run -d -p 3000:3000 -v ./niilearn-data:/app/data niiknow/niilearn

Docker Compose示例

yaml
version: '3'
services:
  niilearn:
    image: niiknow/niilearn
    ports:
      - "3000:3000"
    volumes:
      - ./niilearn-data:/app/data
    restart: always