imsl.data_mining.QUESTDecisionTree.predict

QUESTDecisionTree.predict(data, weights=None)

Compute predicted values using a decision tree.

Parameters:
  • data ((M,N) array_like) – Array containing the data.
  • weights ((M,) array_like, optional) –

    Array containing the case weights.

    Default is weights[i] = 1.0.

Returns:

  • A named tuple with the following fields
  • predictions ((M,) ndarray) – Array containing the predicted values.
  • pred_err_ss (float) – The prediction error mean sum of squares.