Hi Klaus, you can use new ml api with dataframe:
val model = (new
LogisticRegresion).setInputCol("fetures").setProbabilityCol("probability").setOutputCol("prediction").fit(data)
Thanks,
Peter Rudenko
On 2015-06-30 14:00, Klaus Schaefers wrote:
Hello,
is there a way to get the during the predict() phase also the class
probabilities like I would get in sklearn?
Cheers,
Klaus
--
--
Klaus Schaefers
Senior Optimization Manager
Ligatus GmbH
Hohenstaufenring 30-32
D-50674 Köln
Tel.: +49 (0) 221 / 56939 -784
Fax: +49 (0) 221 / 56 939 - 599
E-Mail: klaus.schaef...@ligatus.com
Web: www.ligatus.de
HRB Köln 56003
Geschäftsführung:
Dipl.-Kaufmann Lars Hasselbach, Dipl.-Kaufmann Klaus Ludemann,
Dipl.-Wirtschaftsingenieur Arne Wolter
