Method: Rumale::Ensemble::AdaBoostClassifier#predict_proba

Defined in:
lib/rumale/ensemble/ada_boost_classifier.rb

#predict_proba(x) ⇒ Numo::DFloat

Predict probability for samples.

Parameters:

  • x (Numo::DFloat) —

    (shape: [n_samples, n_features]) The samples to predict the probailities.

Returns:

  • (Numo::DFloat) —

    (shape: [n_samples, n_classes]) Predicted probability of each class per sample.



165
166
167
168
169
170
171
172
 
# File 'lib/rumale/ensemble/ada_boost_classifier.rb', line 165

def predict_proba(x)
  check_sample_array(x)
  n_classes = @classes.size
  probs = Numo::NMath.exp(1.fdiv(n_classes - 1) * decision_function(x))
  sum_probs = probs.sum(1)
  probs /= Numo::DFloat[sum_probs].transpose
  probs
end