每个节点都有二元分类器的“DeepTree”网络

数据挖掘 机器学习 美国有线电视新闻网 决策树
2022-02-19 21:04:56

我需要做一个项目来建立一个“DeepTree Network”。假设我有一个 A、B 和 C 类的 3 分类问题。

我的树应该是这样的:

  • 第一个节点:CNN binary classifier在 C 类和超类 A+B 之间进行分类

  • 第二个节点:CNN binary classifier在 A 类和 B 类之间进行分类

有谁知道是否有关于这个主题的现有论文?或者任何实现?叶子CNNs不一定是,任何像这样的树实现TensorFlow都会对我有很大帮助。

2个回答

您可以使用您的决策功能尝试 SVM:

https://scikit-learn.org/stable/modules/generated/sklearn.svm.SVC.html#sklearn.svm.SVC

它可以使用默认的 ovr 决策函数处理多分类问题

如果你坚持使用 tensorflow,你只能尝试 linver svc:https ://www.tensorflow.org/api_docs/python/tf/contrib/learn/SVM

这似乎足够专业,不太可能有现有的实现。用手做应该很容易。如果您确实需要,可以将其构建为可以适合现有实现的堆叠集成。