kairntech/pysuggester-cpu-sklearn Docker 镜像 - 轩辕镜像

CPU Sklearn 建议器 Docker 镜像文档

1. 镜像概述

本镜像为基于CPU环境的Scikit-learn建议器工具容器化实现，旨在简化机器学习工作流中的模型选择、参数调优及数据预处理决策过程。通过集成轻量级建议算法，可为各类Scikit-learn相关任务提供智能化推荐，支持在无GPU资源的环境中高效部署与运行，适用于模型开发、教学演示及自动化机器学习流程集成场景。

2. 核心功能与特性

2.1 核心功能

• 模型智能推荐：根据任务类型（分类/回归/聚类）及数据集特征（样本量、特征数、数据类型）推荐最优Scikit-learn模型
• 参数调优建议：针对推荐模型提供初始参数范围及调优方向（如正则化强度、树深度等）
• 预处理步骤推荐：基于数据特征自动生成数据清洗、特征工程及归一化/标准化方案
• 兼容性验证：检查建议模型与当前Scikit-learn版本的兼容性，避免版本冲突问题

2.2 关键特性

• CPU优化：专为CPU环境设计，资源占用低，支持在低配设备上运行
• 轻量级部署：镜像体积精简，启动速度快，适合快速集成到现有工作流
• API接口支持：提供RESTful API便于外部系统调用建议服务
• 可配置性：支持通过环境变量自定义建议算法策略及输出格式

3. 使用场景与适用范围

3.1 目标用户

• 数据科学家与机器学习工程师
• 机器学习初学者及教学场景
• 自动化机器学习（AutoML）流程开发者

3.2 典型应用场景

• 模型开发初期：快速确定适合当前数据集的基础模型及参数范围
• 教学演示：直观展示不同模型的适用条件及参数影响
• 生产环境集成：作为AutoML pipeline的前置建议模块，为模型训练提供基础配置
• 边缘计算场景：在无GPU资源的边缘设备上提供轻量级建议服务

4. 使用方法与配置说明

4.1 镜像获取

从Docker仓库拉取最新版本镜像：

bash
docker pull [镜像仓库地址]/cpu-sklearn-suggester:latest

注：请将[镜像仓库地址]替换为实际镜像仓库路径（如Docker Hub用户名或私有仓库地址）

4.2 快速启动

使用默认配置启动容器（映射8080端口提供API服务）：

bash
docker run -d -p 8080:8080 --name sklearn-suggester [镜像仓库地址]/cpu-sklearn-suggester:latest

• -d：后台运行容器
• -p 8080:8080：端口映射（主机端口:容器端口）
• --name：指定容器名称便于管理

4.3 环境变量配置

通过环境变量自定义服务行为，常用配置项如下：

配置示例（使用高级建议模式）：

bash
docker run -d -p 8080:8080 \
  -e LOG_LEVEL=DEBUG \
  -e SUGGEST_MODE=advanced \
  -e MAX_CANDIDATES=3 \
  --name sklearn-suggester [镜像仓库地址]/cpu-sklearn-suggester:latest

4.4 API接口使用

容器启动后，可通过HTTP POST请求调用建议服务，以下为核心接口说明：

4.4.1 获取模型推荐

请求地址：/suggest/model
请求方法：POST
请求体：

json
{
  "task_type": "classification",  // 任务类型：classification/regression/clustering
  "data_info": {
    "num_samples": 5000,           // 样本数量
    "num_features": 15,            // 特征数量
    "feature_types": ["numerical", "categorical"],  // 特征类型列表
    "has_missing_values": true,    // 是否存在缺失值
    "target_distribution": "imbalanced"  // 目标变量分布（仅分类任务）
  }
}

响应示例：

json
{
  "recommended_models": [
    {
      "name": "RandomForestClassifier",
      "confidence": 0.89,
      "reason": "适用于中等规模数据集，对缺失值不敏感，能处理混合类型特征",
      "initial_params": {
        "n_estimators": [50, 200],
        "max_depth": [None, 10, 20]
      }
    },
    {
      "name": "XGBClassifier",
      "confidence": 0.85,
      "reason": "对不平衡数据表现良好，训练速度快",
      "initial_params": {
        "learning_rate": [0.01, 0.1],
        "n_estimators": [100, 300]
      }
    }
  ],
  "preprocessing_suggestions": [
    "缺失值处理：使用SimpleImputer(strategy='median')填充数值特征",
    "分类特征编码：使用OneHotEncoder处理低基数分类特征",
    "特征缩放：对数值特征应用StandardScaler"
  ],
  "compatibility_check": "all recommended models are compatible with scikit-learn>=0.24.0"
}

4.5 数据持久化与扩展

通过挂载数据卷实现配置持久化或日志存储：

bash
docker run -d -p 8080:8080 \
  -v ./local_config:/app/config \  # 挂载本地配置目录
  -v ./local_logs:/app/logs \      # 挂载本地日志目录
  --name sklearn-suggester [镜像仓库地址]/cpu-sklearn-suggester:latest

5. 注意事项

• 本镜像仅支持CPU运行，不提供GPU加速功能
• 建议根据数据规模调整容器CPU资源分配（使用--cpus参数限制CPU核心数）
• 首次启动时可能需要下载基础模型元数据（约50-100MB），请确保网络通畅
• 支持的Scikit-learn最低版本为0.24.0，使用前请确认环境兼容性
• 高并发场景下建议配合负载均衡器使用，避免单容器压力过大