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

Suranga Kasthurirathne surangakas at gmail.com
Fri Oct 28 08:13:31 EDT 2016


Thanks Piotr, this was indeed the case. Works for me now :)

On Wed, Oct 26, 2016 at 11:26 AM, Suranga Kasthurirathne <
surangakas at gmail.com> wrote:

>
> 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
>



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


More information about the scikit-learn mailing list