项目作者: plai-group

项目描述 :
Canonical Correlation Forests (CCFs)
高级语言: Python
项目地址: git://github.com/plai-group/ccfs-python.git
创建时间: 2020-02-12T22:02:12Z
项目社区:https://github.com/plai-group/ccfs-python

开源协议:Apache License 2.0

下载


arxiv badge

Canonical Correlation Forests (CCFs)

CCFs are a decision tree ensemble method for classification and regression. CCFs naturally
accommodate multiple outputs, provide a similar computational complexity to random forests,
and inherit their impressive robustness to the choice of input parameters.

This implementation is completely done using Numpy and SciPy, which are open-sourced
numerical computing libraries.

alt-text-1 alt-text-2
CCFs results on Spiral Dataset CCFs results on Camel Dataset

Setup

  1. pip3 install -e .

(This code base was developed on Python3.6)

Run Examples

For classification example run the following command:

  1. cd tests
  2. python3 classification_example.py

For regression example run the following command:

  1. python3 regression_example.py

Contribution

Any improvements to the code base are welcomed, submit a merge request.

Original Code

  1. https://github.com/twgr/ccfs

Paper citation

  1. @article{rainforth2015canonical,
  2. title={Canonical correlation forests},
  3. author={Rainforth, Tom and Wood, Frank},
  4. journal={arXiv preprint arXiv:1507.05444},
  5. year={2015}
  6. }

alt-text-1