qiskit-community / qiskit-machine-learning / 12344712119

91%

push

github

Add predict_proba Support to PegasosQSVC and NeuralNetworkClassifier (#871)

* Adding a predict_proba function to classifiers. (#57)

* Update README.md

* Predict proba for NNC and PegQSVC

* Rewriting predict proba features and docstring

It was very inefficient before and didn't have the validation checks needed. The code is now more clear and docstring has been added.

* Tweak documentation for NNC and PegasosQSVC, silence lint E1101 on torch connector

* Update test with `QNN.predict_proba`

* Update test with `PegasosESVC.predict_proba`

* Added a release note and solved conflicts with main

---------

Co-authored-by: FrancescaSchiav <FrancescaSchiav@users.noreply.github.com>
Co-authored-by: oscar-wallis <108736468+oscar-wallis@users.noreply.github.com>
Co-authored-by: Edoardo Altamura <38359901+edoaltamura@users.noreply.github.com>
Co-authored-by: smens <88490989+smens@users.noreply.github.com>

* Reformatted docs

* Fix usage of sklearn

---------

Co-authored-by: FrancescaSchiav <FrancescaSchiav@users.noreply.github.com>
Co-authored-by: oscar-wallis <108736468+oscar-wallis@users.noreply.github.com>
Co-authored-by: Edoardo Altamura <38359901+edoaltamura@users.noreply.github.com>
Co-authored-by: smens <88490989+smens@users.noreply.github.com>

15 of 16 new or added lines in 2 files covered. (93.75%)

4364 of 4811 relevant lines covered (90.71%)

0.91 hits per line