如何在 sklearn 库中包含一个新函数

How to include a new function in sklearn library

我希望在一种新型分类模型上编写一个函数并将其包含在 sklearn 中(这样我就可以与全世界分享它!)。我该怎么做呢?是否有 link 显示我应该遵循的所有步骤和指南?我希望以类似于 "form" 的方式编写函数以与 sklearn 中的现有分类模型(例如 .fit(X,Y)、.predict(X) 等)相媲美,以便 "form" 标准化。

sklearn 提供了一个 API 可能包含您正在寻找的内容。

您可以在以下位置找到详细信息:https://scikit-learn.org/stable/developers/develop.html