如何在scikit-learn中添加新类并使用它

时间:2019-07-01 14:30:39

标签: python scikit-learn

我以开发人员模式安装了sci-learn,因为我想对其进行修改以供自己使用。 (如果我的申请成功,有用并得到社区的认可,我将考虑为sklearn做出贡献)

我想向sklearn.tree添加一个称为MultiTree的新类,该类可以返回一组满足某些条件的树。我所做的就是将新类直接附加到/scikit-learn/sklearn/tree/tree.py。但是我无法导入添加的新类。

我已将新类添加到__all__文件夹的tree.py__init__.py的{​​{1}}变量中。

我使用命令/scikit-learn/sklearn/tree安装了scikit-learn软件包,我认为这样可以添加新代码。但我认为我错了。

此外,我不知道我的方法是否是修改sklearn供我使用的正确方法。如果没有,请随时提出正确的方法。

谢谢您的帮助!

0 个答案:

没有答案