北数云|利用Limix模型对tabular-benchmark数据集实现分类和回归任务

利用Limix模型对tabular-benchmark数据集实现分类和回归任务

LimiX 是一个专为结构化数据设计的轻量级大模型(仅约 16M 参数),无需任何下游训练即可直接推理,同时具备优秀的泛化能力与概率校准性能。在多种分类与回归任务中,其无训练模式已能达到或超越传统表格模型基线,表现稳健。结合可选的检索增强推理机制(Retrieval Ensemble),模型能够进一步利用训练样本关系改善复杂分布下的预测精度。整体而言,LimiX 在推理成本与效果之间实现了高效平衡,适合在低训练资源场景中快速获得可靠结果。

创建项目

点击工作台右上角 创建项目 新建项目。

初始化配置

按照下图推荐配置,选择镜像、数据及模型

资源配置:P1.large及以上,算力直接影响到生成结果语音的速度,若您选择更高配置,请您确保您有足够的算力点

镜像:选择LimiX-1028

数据:选择LimiX-code、tabular-benchmark

模型:暂无

代码拷贝

cd到LimiX目录下,执行以下指令,将代码拷贝到 /gemini/code 目录下,将数据集拷贝到 /gemini/code/data 目录下

cp -r /gemini/data-1/* /gemini/code
cp -r /gemini/data-2/tabular-benchmark /gemini/code/data

执行完后,可使用 ls命令 检查是否成功,如图所示

数据预处理

利用 process_data.py 脚本对 tabular-benchmark 进行数据预处理,以便LimiX模型进行推理

cd /gemini/code/LimiX-main # 进入启动目录

处理分类数据集示例

python process_data.py \
--input_dir /gemini/code/data/tabular-benchmark/clf_cat \
--output_dir /gemini/code/data/tabular-benchmark_clf_cat_split \
--task_type classification

处理回归数据集示例

python process_data.py \
--input_dir /gemini/code/data/tabular-benchmark/reg_cat \
--output_dir /gemini/code/data/tabular-benchmark_reg_cat_split \
--task_type regression

以上代码仅为示例,只对 clf_cat 与 reg_cat 两组数据进行处理,您也可自行更换输入输出路径,请标明任务类型 task_type 为 classification(分类)或 regression(回归)

模型推理

利用 inference_classifier.py 和 inference_regression.py 脚本分别执行分类和回归任务

cd /gemini/code/LimiX-main # 进入启动目录

执行分类任务

python inference_classifier.py \
--save_name cls_clfcat_noretrieval \
--inference_config_path config/cls_default_noretrieval.json \
--data_dir /gemini/code/data/tabular-benchmark_clf_cat_split \
--model_path /gemini/pretrain/LimiX-16M/LimiX-16M.ckpt

处理回归数据集示例

python inference_regression.py \
--save_name reg_clfcat_noretrieval \
--inference_config_path config/reg_default_noretrieval.json \
--data_dir /gemini/code/data/tabular-benchmark_reg_cat_split \
--model_path /gemini/pretrain/LimiX-16M/LimiX-16M.ckpt

以上代码仅为示例,只对处理过的 clf_cat 与 reg_cat 数据进行处理,模式为无需检索模式(notrieval),速度更快,所需性能和内存较低,

查看推理结果

利用JupyterLab打开 /gemini/code/LimiX-main/result 文件夹,查看刚刚分类和回归的输出,指标保存在 all_rst.csv 中

温馨提示: 完成项目后,记得及时关闭开发环境,以免继续产生费用。如果担心自己忘记关闭,可以提前在右边栏配置处设置自动停止时间。

北数云正在加速接入

更多"科研级"前沿模型

我们相信,通用大模型的普惠化,不仅是技术进步,更是科研创新生态的重要推动力。未来,北数云还将持续上线更多前沿与工具。

欢迎各位用户留言告诉我们:你最希望在北数云看到哪些模型?

我们会持续关注大家的实际科研需求,把算力与模型能力结合,助力每一位开发者和研究者,以最低成本释放最大潜能!

高校专属算力普惠计划进行中!完成认证领1000算力点

相关推荐
想会飞的蒲公英3 分钟前
TF-IDF + 随机森林中文文本分类全链路实战:从训练脚本到 Flask API + Streamlit 前端
人工智能·pytorch·python·随机森林·分类·flask·tf-idf
轩渃1 天前
Claude Fable5回归即翻车:跑分暴跌、安全拦截、账单猫腻,AI模型的信任危机
人工智能·安全·回归
二狗的编程之路1 天前
神经网络正则化:防止过拟合的七种核心手段
python·机器学习·数据挖掘
风流 少年2 天前
数据分析:numpy
数据挖掘·数据分析·numpy
漂亮的摩托2 天前
基于OpenCV与SVM的车牌识别系统实现:定位、分割与分类全流程解析
opencv·支持向量机·分类
大鱼>2 天前
机器学习基础:从零理解核心概念与算法分类
算法·机器学习·分类
AI科技星2 天前
基于32维Cayley_Dickson超复数的全域拓扑统一场论——反重力、真空自持供能、维度瞬移与星际宇宙脑秩序体系
人工智能·学习·算法·机器学习·数据挖掘
AI科技星2 天前
生命几何学:数学分子生物学讲义(中文定稿+完整可编译LaTeX双版终稿)
人工智能·学习·算法·机器学习·数据挖掘·乖乖数学·全域数学
hu9245195592 天前
滞回曲线还在手动处理?一键滞回曲线平滑,提取骨架曲线、延性系数、耗能面积,批量搞定拟静力试验数据分析
数据挖掘·数据分析
YangYang9YangYan2 天前
2026大数据专业毕业学数据分析的价值
大数据·数据挖掘·数据分析