BriefGPT - AI 论文速递 · 2024-10-28T00:00:00Z

skwdro：一个计算瓦瑟斯坦分布鲁棒机器学习的库

📝

内容提要

本文介绍了skwdro，一个用于训练鲁棒机器学习模型的Python库，旨在解决模型在分布变化时的脆弱性。该库兼容scikit-learn和PyTorch，采用最优传输距离的分布鲁棒优化方法，提高了模型的灵活性和易用性。

🏷️

标签

skwdro 优化方法 分布变化 机器学习 鲁棒性