skorch 1.1.0发布:强化scikit-learn兼容性与学习率调度功能
skorch是一个基于PyTorch的scikit-learn兼容神经网络库,它允许开发者使用熟悉的scikit-learn API来训练PyTorch模型。这个库弥合了深度学习框架PyTorch与传统机器学习库scikit-learn之间的鸿沟,使得PyTorch模型能够无缝集成到scikit-learn的工作流中。
核心更新内容
scikit-learn 1.6.0兼容性增强
skorch 1.1.0最重要的改进之一是全面支持scikit-learn 1.6.0及更高版本。所有神经网络类现在都继承自scikit-learn的BaseEstimator基类,分类模型额外继承ClassifierMixin,回归模型继承RegressorMixin。这一改变确保了skorch模型能够完全兼容最新的scikit-learn生态系统,包括管道(Pipeline)、网格搜索(GridSearchCV)等工具。
学习率调度功能改进
新版本对学习率调度器进行了多项增强:
-
当使用ReduceLROnPlateau调度器时,现在会默认记录学习率变化到网络历史中(通过net.history[:, 'event_lr']访问),这为模型训练过程提供了更详细的监控能力。
-
学习率调度现在支持按批次(batch)更新,而不仅仅是按周期(epoch)更新,这为精细控制训练过程提供了更多灵活性。
-
调度器的simulate()方法现在支持添加步骤参数,这对于模拟ReduceLROnPlateau等需要基于指标调整学习率的策略特别有用。
新增学习率调度示例
为了帮助用户更好地理解和使用学习率调度功能,项目新增了一个Jupyter Notebook示例,详细展示了如何在skorch中使用各种学习率调度策略。这个示例对于理解如何优化模型训练过程非常有价值。
问题修复
-
修复了NeuralNetBinaryClassifier与torch.compile的兼容性问题,这使得二进制分类模型现在可以正确使用PyTorch 2.0的编译优化功能。
-
移除了已弃用的skorch.callbacks.scoring.cache_net_infer功能,保持代码库的整洁性。
技术意义与应用价值
skorch 1.1.0的这些改进使得PyTorch模型能够更好地融入scikit-learn生态系统,特别是在以下方面:
-
对于需要将深度学习模型与传统机器学习模型结合使用的场景,如集成学习或模型堆叠(stacking),兼容性的提升使得工作流程更加顺畅。
-
学习率调度功能的增强为模型训练提供了更精细的控制能力,有助于提升模型性能并加速收敛。
-
新增的示例文档降低了学习曲线,使得新用户能够更快上手高级功能。
这些改进共同提升了skorch在工业生产环境中的适用性,使其成为连接PyTorch深度学习能力和scikit-learn机器学习工作流的理想桥梁。
atomcodeClaude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed. Get StartedRust0197
cann-learning-hubCANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。Jupyter Notebook0126
MiMo-V2.5-Pro-FP4-DFlashMiMo-V2.5-Pro-FP4-DFlash 是驱动 MiMo-V2.5-Pro-UltraSpeed 的底层模型: FP4 量化骨干网络:对 MoE 专家采用 MXFP4 量化,同时保持模型其他部分的更高精度,在几乎无损质量的前提下,显著减小模型体积并降低内存带宽压力。 BF16 DFlash 草稿生成器:用于块扩散推测解码,每次前向传播可生成一整个块的 tokens,并让骨干网络一步完成验证。 两者协同作用,既降低了每参数的位宽,又减少了骨干网络前向传播的次数,而这两者正是万亿参数模型解码过程中的两大主要成本来源。Python00
JoyAI-EchoJoyAI-Echo,这是一个独立的、仅用于推理的版本,旨在实现分钟级多镜头音视频生成。它采用了经过蒸馏的DMD生成器、配对的跨模态记忆以及故事级别的一致性。其性能的核心在于,一个跨模态视听记忆库能够在长达五分钟的视频中保持角色外观和语音音色的一致性。同时,一个训练后处理流程将基于记忆的强化学习与分布匹配蒸馏相结合,实现了7.5倍的速度提升,显著增强了视觉质量和对齐效果。00
AstrBot✨ 易上手的多平台 LLM 聊天机器人及开发框架 ✨ 平台支持 QQ、QQ频道、Telegram、微信、企微、飞书 | OpenAI、DeepSeek、Gemini、硅基流动、月之暗面、Ollama、OneAPI、Dify 等。附带 WebUI。Python06
handy-ollama动手学Ollama,CPU玩转大模型部署,在线阅读地址:https://datawhalechina.github.io/handy-ollama/Jupyter Notebook07