讲师中心 微信公众号
AI工具推荐 视频效率加速

如何在 PyCharm 中使用 CodeGeex 快速实现常用的机器学习预处理算法

冬强小哥_7960

冬强小哥_7960

发布时间:2026-06-15 20:16:18

|

848人浏览过

|

来源于php中文网

原创

CodeGeeX插件可直接根据自然语言描述生成sklearn预处理代码:支持标准化/归一化、缺失值填充、OneHot编码、PCA降维、Tfidf+特征选择等,自动生成含ColumnTransformer和Pipeline的完整可运行代码,并确保fit_transform仅用于训练集。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

如何在 pycharm 中使用 codegeex 快速实现常用的机器学习预处理算法

你需要在 PyCharm 中快速把原始数据变成模型可训练的格式,比如标准化、编码分类变量、处理缺失值,但每次都要翻文档查 sklearn 参数、写 pipeline、反复调试 fit/transform 顺序——CodeGeeX 能直接根据你的自然语言描述生成可运行的预处理代码。

安装并激活 CodeGeeX 插件

打开 PyCharm → File → Settings → Plugins → Marketplace 搜索 【CodeGeeX】 → 点击 Install → 安装完成后重启 IDE。

重启后右下角出现 CodeGeeX 图标且状态为绿色,表示插件已就绪;若图标灰显,需点击登录并绑定账号(未注册则跳转 codegeex.cn 完成注册)。

用注释触发标准化与归一化代码生成

在 Python 文件中新建空函数,光标置于函数体内,输入以下注释:

# 对数值特征['age', 'income', 'score']做Z-score标准化;对['salary_range', 'education_level']做Min-Max归一化到[0,1]区间;保留原始DataFrame结构,不修改列名

按下 Ctrl+Enter(Windows/Linux)或 Cmd+Enter(Mac) 触发生成。等待右下角图标停止旋转,灰色代码块出现后按 Tab 键插入。

生成代码会自动导入 StandardScaler 和 MinMaxScaler,并使用 ColumnTransformer 分别作用于不同列组——这一步不能手动拼接两个 scaler,否则 transform 时会报错“n_features_in_ mismatch”。

一键生成带缺失值策略的完整预处理 Pipeline

方法一:自然语言指令直出

在空行输入:# 构建sklearn Pipeline:先用SimpleImputer(strategy='most_frequent')填充类别列缺失值,再用OneHotEncoder(drop='first')编码;对数值列用median填充后StandardScaler;最后用PCA(n_components=0.95)降维

触发生成后,检查是否包含 Pipeline、ColumnTransformer、make_column_selector 的组合调用;若生成代码中 PCA 前缺少 scaler,说明提示词未强调“数值列先缩放再降维”,需补全“PCA 必须作用于已标准化的数据”再次生成。

方法二:基于已有 DataFrame 补全

PyCharm 2026.2
PyCharm 2026.2

PyCharm 2026.2是 JetBrains PyCharm 的指定版本安装包,下载地址指向官方 Windows 安装包直链,可用于旧项目兼容、版本回退和环境测试。

下载

确保当前文件中已定义 df = pd.read_csv("data.csv"),且列名清晰可见。选中该行 → 右键 → CodeGeeX → “Generate Preprocessing Code for This DataFrame” → 在弹出框中追加:“目标是用于后续 LogisticRegression 训练,要求输出 X_train, X_test, y_train, y_test 划分结果”。

生成代码将自动注入 train_test_split,并对 X 做完整预处理、y 做 LabelEncoder ——注意它不会擅自修改原始 df,所有操作都在副本上进行。

处理文本特征:Tfidf + 特征选择联动生成

第一步:明确输入结构

在代码中先写好基础读取逻辑:df = pd.read_csv("reviews.csv"); text_col = "review_text"; label_col = "sentiment"

第二步:输入精准提示

# 对 review_text 列提取 Tfidf 特征(max_features=5000, ngram_range=(1,2)),然后用 SelectKBest(chi2, k=1000) 筛选最具区分度的词项,最终与数值列 ['rating', 'helpful_votes'] 拼接成完整特征矩阵

第三步:执行生成并验证关键点

生成代码必须包含 FeatureUnion 或 ColumnTransformer 将文本与数值特征合并;若只返回了 TfidfVectorizer 单独调用,说明提示词中“拼接”一词未被识别,应改为“horizontal stack with numerical columns”重新触发。

第四步:检查 fit_transform 调用位置

【必须仅对训练集调用 fit_transform,测试集只能用 transform】 ——生成代码若在 test 部分也写了 fit_transform,会引发 Data Leakage,需手动删掉 test 部分的 fit_ 前缀。

热门AI工具

更多
SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

1751

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

4524

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1729

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

26597

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3227

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3247

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

1183

2023.08.11

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

596

2023.08.10

OpenJDK JVM参数怎么调优
OpenJDK JVM参数怎么调优

Java应用运行缓慢,如何调整JVM参数?本专题介绍OpenJDK JVM性能调优的基本流程,涵盖性能指标确认、运行数据采集、内存配置和垃圾回收分析,帮助你先定位瓶颈,再通过对照测试验证效果,避免直接套用参数而影响应用稳定性。

20

2026.10.16

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn