[scikit-learn] Calculating prediction probability per each predicted outcome

Suranga Kasthurirathne surangakas at gmail.com
Wed Oct 26 11:26:20 EDT 2016


Hi everyone,

I'm currently using Scikit learn to train and test multiple neural networks.

My issue - I'm breaking my dataset into 90/10, training on the 90%, and
testing on the 10%.

For the 10% trained data, I get outcomes as follows:

predicted = neural_network.predict(test_data)

Here, the predicted variable is basically either 1 or 0, which is what i'm
feeding in as the outcome.

But how can I get the prediction probability per each predicted outcome?
back in the day when I used weka it produced a single prediction, followed
by a prediction probability between 1 and 0 for each outcome.


-- 
Best Regards,
Suranga
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/scikit-learn/attachments/20161026/62ae0d41/attachment.html>


More information about the scikit-learn mailing list