Commit 6bb60b38 authored by Rakesh MEHTA's avatar Rakesh MEHTA
Browse files

return score along with label

parent 445d6a75
......@@ -270,6 +270,8 @@ class BoostMachine():
Return:
prediction_scores: The real valued number which are thresholded to determine the prediction classes.
prediction_labels: The predicted classes for the test samples. It is a binary numpy array where
1 indicates the predicted class.
Type: numpy array
......@@ -305,7 +307,7 @@ class BoostMachine():
else:
score_max = numpy.argmax(pred_scores, axis = 1)
pred_labels[range(num_samp),score_max] = 1
return pred_labels
return pred_scores, pred_labels
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment