Machine Learning for OpenCV 4 - Second Edition

Using decision trees for regression

Although we have so far focused on using decision trees in classification tasks, you can also use them for regression. But you will need to use scikit-learn again, as OpenCV does not provide this flexibility. We will therefore only briefly review its functionality here:

1. Let's say we wanted to use a decision tree to fit a sin wave. To make things interesting, we will also add some noise to the data points using NumPy's random number generator:

In [1]: import numpy as np...     rng = np.random.RandomState(42)

2. We then create 100 randomly spaced x values between 0 and 5 and calculate the corresponding sin values:

In [2]: X = np.sort(5 * rng.rand(100, 1), axis=0)...     y = np.sin(X).ravel()

3. We then add noise to ...