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

时间:2015-08-24 11:35:32

标签: python scikit-learn

我想将factorization machines整合到sklearn中。我检查了sklearn文档和Web以了解如何包装新算法,但这个要求似乎没有很好地记录。

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

2 个答案:

答案 0 :(得分:5)

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

XGBoost模块有一个完整的sklearn包装器,您可以在此处看到:

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

答案 1 :(得分:0)

从这个常见问题中我得到的是他们不太喜欢新算法http://scikit-learn.org/stable/faq.html#selectiveness,原因似乎是有效的。鉴于此,认为没有关于如何添加新算法包装器的任何文档是合理的。我将补充一点,我现在已经使用该软件包了一段时间,而且我从未在他们的网站或其他类似于您正在寻找的网站上找到任何内容。