scikit-learn-tips

GitHub
1.7k 437 非常简单 1 次阅读 2天前数据工具其他开发框架
AI 解读 由 AI 自动生成,仅供参考

scikit-learn-tips 是一个专为提升机器学习实战能力而设计的开源学习资源库,由数据科学专家 Just Markham 精心整理。它汇集了 50 个针对 Python 主流机器学习库 scikit-learn 的实用技巧,旨在帮助用户解决从数据预处理到模型构建过程中的常见痛点与误区。

许多开发者在使用 scikit-learn 时,往往对核心概念(如 fit 与 transform 的区别)、复杂流程(如使用 ColumnTransformer 对不同列应用不同处理)以及类别特征编码等细节感到困惑,导致代码效率低下或出现数据泄露风险。scikit-learn-tips 通过短小精悍的视频教程、可交互的 Jupyter Notebook 代码示例以及社交媒体上的深入讨论,将这些抽象概念转化为直观易懂的操作指南。例如,它不仅演示了如何正确处理训练集与测试集的转换逻辑,还对比了使用 scikit-learn 而非 pandas 进行预处理的专业优势。

这套资源特别适合具有一定 Python 基础的数据科学家、机器学习工程师以及正在进阶的学习者。无论你是希望规范代码习惯的开发者,还是想要深入理解算法底层逻辑的研究人员,都能从中获得启发。其独特的“视频 + 代码 + 社区互动”三位一体学习模式,让枯燥的技术细节变得生动有趣,是掌握 scikit-learn 最佳实践的高效途径。

使用场景

某电商数据分析师正在构建用户流失预测模型,面对包含数值型消费记录和分类型用户标签的混合数据集,急需高效完成特征工程。

没有 scikit-learn-tips 时

  • 手动编写繁琐的 Pandas 代码分别处理不同列,导致预处理逻辑分散且难以维护,容易出错。
  • 混淆 fittransformfit_transform 的使用场景,错误地在测试集上执行 fit,造成严重的数据泄露。
  • 遇到训练集中未出现的新类别(未知分类)时,模型直接报错崩溃,缺乏优雅的容错机制。
  • 无法通过统一管道串联预处理与建模步骤,每次调整参数都需重复执行大量中间代码,效率极低。

使用 scikit-learn-tips 后

  • 利用 ColumnTransformer 一键定义不同列的差异化处理策略,将杂乱的预处理逻辑封装为简洁、可复用的管道。
  • 清晰掌握“训练集用 fit_transform,测试集仅用 transform"的核心原则,彻底杜绝数据泄露风险,确保评估结果真实可靠。
  • 配置 OneHotEncoderhandle_unknown='ignore' 参数,轻松化解未知类别导致的程序中断,提升模型鲁棒性。
  • 借鉴最佳实践将预处理与算法模型无缝集成,实现从原始数据到预测结果的全流程自动化,大幅缩短迭代周期。

scikit-learn-tips 通过传授标准化的实战技巧,帮助开发者将碎片化的预处理操作转化为严谨、高效的机器学习工作流。

运行环境要求

GPU

未说明

内存

未说明

依赖
notes该项目主要是一系列关于 scikit-learn 使用技巧的 Jupyter Notebook 教程集合,并非一个需要复杂安装部署的软件工具。运行这些示例代码通常需要安装 scikit-learn、pandas 以及用于查看代码的 Jupyter Notebook 环境。具体的 Python 版本和库版本需参考各个 Notebook 文件内部的代码兼容性,README 中未明确指定最低系统资源或特定版本要求。
python未说明
scikit-learn
pandas
jupyter
scikit-learn-tips hero image

快速开始

🤖⚡ scikit-learn 小贴士

新小贴士会发布在 LinkedInTwitterFacebook 上。

👉 立即注册,每周通过电子邮件收到 2 条视频小贴士! 👈

所有小贴士列表

点击 LinkedIn 上讨论该小贴士,点击 查看某条小贴士的 Jupyter 笔记本,或点击 YouTube 上观看该小贴士的视频:

# Description Links
1 Use ColumnTransformer to apply different preprocessing to different columns
2 Seven ways to select columns using ColumnTransformer
3 What is the difference between "fit" and "transform"?
4 Use "fit_transform" on training data, but "transform" (only) on testing/new data
5 Four reasons to use scikit-learn (not pandas) for ML preprocessing
6 Encode categorical features using OneHotEncoder or OrdinalEncoder
7 Handle unknown categories with OneHotEncoder by encoding them as zeros
8 Use Pipeline to chain together multiple steps
9 Add a missing indicator to encode "missingness" as a feature
10 Set a "random_state" to make your code reproducible
11 Impute missing values using KNNImputer or IterativeImputer
12 What is the difference between Pipeline and make_pipeline?
13 Examine the intermediate steps in a Pipeline
14 HistGradientBoostingClassifier natively supports missing values
15 Three reasons not to use drop='first' with OneHotEncoder
16 Use cross_val_score and GridSearchCV on a Pipeline
17 Try RandomizedSearchCV if GridSearchCV is taking too long
18 Display GridSearchCV or RandomizedSearchCV results in a DataFrame
19 Important tuning parameters for LogisticRegression
20 Plot a confusion matrix
21 Compare multiple ROC curves in a single plot
22 Use the correct methods for each type of Pipeline
23 Display the intercept and coefficients for a linear model
24 Visualize a decision tree two different ways
25 Prune a decision tree to avoid overfitting
26 Use stratified sampling with train_test_split
27 Two ways to impute missing values for a categorical feature
28 Save a model or Pipeline using joblib
29 Vectorize two text columns in a ColumnTransformer
30 Four ways to examine the steps of a Pipeline
31 Shuffle your dataset when using cross_val_score
32 Use AUC to evaluate multiclass problems
33 Use FunctionTransformer to convert functions into transformers
34 Add feature selection to a Pipeline
35 Don't use .values when passing a pandas object to scikit-learn
36 Most parameters should be passed as keyword arguments
37 Create an interactive diagram of a Pipeline in Jupyter
38 Get the feature names output by a ColumnTransformer
39 Load a toy dataset into a DataFrame
40 Estimators only print parameters that have been changed
41 Drop the first category from binary features (only) with OneHotEncoder
42 Passthrough some columns and drop others in a ColumnTransformer
43 Use OrdinalEncoder instead of OneHotEncoder with tree-based models
44 Speed up GridSearchCV using parallel processing
45 Create feature interactions using PolynomialFeatures
46 Ensemble multiple models using VotingClassifer or VotingRegressor
47 Tune the parameters of a VotingClassifer or VotingRegressor
48 Access part of a Pipeline using slicing
49 Tune multiple models simultaneously with GridSearchCV
50 Adapt this pattern to solve many Machine Learning problems

您可以使用 Binder 在线交互式地运行所有这些笔记本:点击此处

注意: 有些技巧不包含代码,只能在 LinkedIn 上查看。

这些技巧是由谁创作的?

你好!我是凯文·马卡姆,Data School 的创始人。自 2014 年以来,我一直教授 Python 数据科学课程。我创建这些技巧是因为我非常喜欢使用 scikit-learn,并希望帮助更多人更高效地运用它。

我该如何提升自己对 scikit-learn 的掌握水平?

我开设了三门课程:

👉 立即了解哪门课程最适合您! 👈

您还有其他技巧吗?

有的!2019 年,我发布了 100 个 pandas 技巧。此外,我还制作了一段视频,分享了我的 pandas 前 25 大技巧

© 2020–2021 Data School。保留所有权利。

常见问题

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|昨天
Agent开发框架图像

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|2天前
开发框架图像Agent

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

142.7k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

107.9k|★★☆☆☆|昨天
开发框架图像Agent

markitdown

MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器

93.4k|★★☆☆☆|今天
插件开发框架

LLMs-from-scratch

LLMs-from-scratch 是一个基于 PyTorch 的开源教育项目,旨在引导用户从零开始一步步构建一个类似 ChatGPT 的大型语言模型(LLM)。它不仅是同名技术著作的官方代码库,更提供了一套完整的实践方案,涵盖模型开发、预训练及微调的全过程。 该项目主要解决了大模型领域“黑盒化”的学习痛点。许多开发者虽能调用现成模型,却难以深入理解其内部架构与训练机制。通过亲手编写每一行核心代码,用户能够透彻掌握 Transformer 架构、注意力机制等关键原理,从而真正理解大模型是如何“思考”的。此外,项目还包含了加载大型预训练权重进行微调的代码,帮助用户将理论知识延伸至实际应用。 LLMs-from-scratch 特别适合希望深入底层原理的 AI 开发者、研究人员以及计算机专业的学生。对于不满足于仅使用 API,而是渴望探究模型构建细节的技术人员而言,这是极佳的学习资源。其独特的技术亮点在于“循序渐进”的教学设计:将复杂的系统工程拆解为清晰的步骤,配合详细的图表与示例,让构建一个虽小但功能完备的大模型变得触手可及。无论你是想夯实理论基础,还是为未来研发更大规模的模型做准备

90.1k|★★★☆☆|昨天
语言模型图像Agent