如何为新的ML算法添加sklearn包装器

Moh*_*OUI 5 python scikit-learn

我想在sklearn中集成分解机器.我检查了sklearn文档和Web以了解如何包装新算法,但这个要求似乎没有很好地记录.

所以,我想询问是否有关于如何向sklearn添加新算法包装器的文档(除了阅读源代码)?

inv*_*ion 5

完成sklearn文档后,最好的办法是查看完整的工作示例.

XGBoost模块有一个彻底的sklearn包装,你可以在这里看到:

https://github.com/dmlc/xgboost/blob/master/python-package/xgboost/sklearn.py