登录
推荐 文章 Go 技术 课程 下载 专题 AI
首页 >  科技周边 >  人工智能

scikit-learn DummyClassifier 怎样建立分类基线

来源:17golang原创

时间:2026-10-29 01:38:29 189浏览 收藏

用 scikit-learn DummyClassifier 建立分类基线,可以先在训练集上拟合一个不使用特征值的分类器,再在与候选模型完全相同的留出集上计算相同指标。最小写法是选择 strategy="prior" 后调用 fit 和 predict。它提供的是“不利用特征也能做到什么程度”的参照,不是可直接上线的业务模型。

官方地址:https://scikit-learn.org/stable/modules/generated/sklearn.dummy.DummyClassifier.html

什么时候应该把基线加入评估任务

新建分类任务、改动特征或重新生成数据划分时,都应同步计算基线。如果一组样本中大部分都属于同一类,只预测多数类也可能得到很高准确率;候选模型的单独分数不能说明它真正利用了特征。

评估进程只需读取获准使用的特征和标签,不必拥有生产写入权限。下面使用合成数据,不读取个人信息、不连接业务系统。真实任务应明确样本可用时间和实体分组规则:同一个用户的相关记录不能仅为了方便就随机分散到训练和评估两侧;时间预测任务也不能把未来数据混入训练。

选定策略,固定同一比较口径

DummyClassifier 的所有策略都忽略 X 中的特征值,通常根据训练标签 y 决定预测行为。prior 的类别预测采用训练多数类,概率输出反映训练标签比例;most_frequent 的类别预测相同,但概率形式不同。需要可重复的随机参照时,可以显式选择随机策略并固定 random_state。

策略适用参照注意事项
prior多数类预测及标签先验默认策略应仍显式写出,便于记录
most_frequent只关心多数类标签不能与 prior 的概率输出混用
stratified / uniform随机预测参照固定种子并关注波动
constant固定业务标签参照明确 constant 标签及指标含义

比较口径至少包括训练索引、留出索引、标签定义、样本权重和指标名称。若训练集经过重采样,必须写清基线使用哪一份训练标签,不能无说明地把原始分布基线与另一种口径的分数并排。

训练标签、策略与留出索引之间的静态约束
基线结构说明图:训练标签限定基线,留出索引限定比较对象;连线表示依赖关系,不代表执行时间或实际结果。

用同一留出集生成最小比较记录

以下 Python 示例需要已安装 scikit-learn。它只在内存中生成二分类数据;数据划分和基线都显式记录参数。代码供读者复现,本文不提供虚构运行分数。

import json
import sklearn
from sklearn.datasets import make_classification
from sklearn.dummy import DummyClassifier
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import accuracy_score, balanced_accuracy_score
from sklearn.model_selection import train_test_split
from sklearn.pipeline import make_pipeline
from sklearn.preprocessing import StandardScaler

# 合成不均衡样本,仅演示比较接口,不代表真实业务效果
X, y = make_classification(
    n_samples=600, n_features=8, n_informative=4,
    n_redundant=1, weights=[0.85, 0.15], random_state=29
)
# 一次划分后供两个分类器共同使用;仅适用于独立同分布示例
X_train, X_eval, y_train, y_eval = train_test_split(
    X, y, test_size=0.3, stratify=y, random_state=31
)
models = {
    "baseline": DummyClassifier(strategy="prior"),
    # 标准化也只在训练数据上拟合
    "candidate": make_pipeline(
        StandardScaler(), LogisticRegression(max_iter=1000)
    ),
}
report = {
    "sklearn_version": sklearn.__version__,
    "split_seed": 31,
    "baseline_strategy": "prior",
    "metrics": {},
}
for name, model in models.items():
    # 两个模型都不能使用留出标签参与拟合
    model.fit(X_train, y_train)
    predicted = model.predict(X_eval)
    report["metrics"][name] = {
        "accuracy": float(accuracy_score(y_eval, predicted)),
        "balanced_accuracy": float(
            balanced_accuracy_score(y_eval, predicted)
        ),
    }

# 只输出汇总指标;异常时让进程失败,不生成伪造的成功记录
print(json.dumps(report, ensure_ascii=False, indent=2))

输出记录含版本、划分种子、基线策略和两个模型的指标。这里把标准化放进 Pipeline,避免先对整个数据集拟合变换再划分。迁移到真实数据时,还应保存实际样本索引或划分文件;单有随机种子无法抵消输入排序与数据内容变化。

分数达标与评估失败要分别处理

普通 accuracy 按全部样本的正确比例计分;balanced accuracy 对各类别召回率取平均,更容易暴露只猜多数类的局限。对于同时包含两类的二分类留出集,恒定预测一类时,一类召回率为 1,另一类为 0,因而 balanced accuracy 为 0.5。这是指标定义的推导,不是本例的实测结果。

候选模型超过基线,只说明在这份划分和这些指标上有增益,不能直接推出线上可用。应事先按业务代价确定少数类召回率、精确率或成本约束,并在独立测试数据上做最终判断。不要看完留出分数再把门槛调到刚好通过,也不要反复用最终测试集挑参数。

基线分数、候选分数和评估记录的组成
比较记录说明图:两组分数共享指标定义,版本与划分信息共同描述评估记录;图中不展示性能数值或上线结论。

若类样本太少导致分层划分失败,应调整数据收集或评估设计,不应去掉报错后假装比较成立。若出现缺失值、候选模型训练失败或收敛警告,应在任务中保留失败或警告状态,修正数据与参数后重新评估;不能用零分覆盖异常,让自动报告无法区分“模型差”和“任务没完成”。

把基线作为每次评估的固定附件

每次任务保存数据版本、划分标识、依赖版本、策略、指标和失败原因。只发送汇总变化及记录位置给负责评估的人,避免通知中带入原始样本。自动触发可以来自数据版本更新或模型候选提交,但应使用同一套比较定义,不能每次临时改变评分标准。

DummyClassifier 需要标准化吗?它不利用特征值,不需要为了基线额外学习缩放参数;候选模型所需的预处理仍必须只在训练侧拟合。

为什么 prior 与 most_frequent 的准确率一样?两者类别预测都采用训练多数类;差别主要在概率输出,概率评估时不能忽略。

一次划分就足够吗?最小例子只展示接口。实际任务需要与时间、分组和样本量相匹配的重复评估或交叉验证设计,并将基线放进每个相同训练与评估划分中。

基线的价值是给复杂模型一个透明参照。保留相同划分、相同指标和明确失败状态,才能判断增加训练成本是否换来了可解释的改善。

参考:DummyClassifier 官方说明、balanced_accuracy_score 官方说明。

声明:本文转载于:17golang原创 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>